混合整数和列表

Question

我希望写一个函数foo生成以下的输出：

 
foo(0) -> "01 00" 
foo(1) -> "01 01" 
foo([0]) -> "01 00" 
foo([1]) -> "01 01" 
foo([0,1]) -> "02 00 01" 
foo([0,1,10]) -> "03 00 01 0a" 

你将如何实现这一点？我是否需要明确说明争论的类型？

十六进制（值）[2：]可用于根据需要转换为十六进制。

谢谢！

巴里

Answer 1

如果参数是一个int，使之成为list。然后你可以继续你的列表处理...

>>> def foo(x): 
...  hexify = lambda n: hex(n)[2:].zfill(2) 
...  if not isinstance(x, list): 
...   x = [x] 
...  return hexify(len(x)) + ' ' + ' '.join(map(hexify, x)) 
... 
>>> foo(0) 
'01 00' 
>>> foo(1) 
'01 01' 
>>> foo([0]) 
'01 00' 
>>> foo([1]) 
'01 01' 
>>> foo([0,1]) 
'02 00 01' 
>>> foo([0,1,10]) 
'03 00 01 0a' 

Answer 2

你需要做一些类型检查。这一切都取决于你想要接受的。

通常，您可以在list构造函数中包装一些东西，并从中获取一个列表。

def foo(x): 
    x = list(x) 

但是转换完全取决于list。例如：list({1: 2})不会给你[2]，但[1]。

所以，如果你想保护用户免受意外，你应该检查输入是单个整数还是列表。与iterables

>>> isinstance("hej", int) 
False 
>>> isinstance("hej", (int, list)) 
False 
>>> isinstance([1,2,3], (int, list)) 
True 
>>> isinstance(1, (int, list)) 
True 

另外一个问题，你不能保证每一个成员是同类型的，例如：

你可以用isinstance检查这

[1, 'hello', (2.5,)] 

我只是想尝试把每个项目，如果不可能，请将双手举起，并向使用者抱怨。

Answer 3

def tohex(n): 
    ''' Convert an integer to a 2-digit hex string. ''' 
    hexVal = hex(n)[2:] 
    return hexVal.rjust(2, '0') 

def foo(input): 
    # If we don't get a list, wrap whatever we get in a list. 
    if not isinstance(input, list): 
     input = [input] 

    # First hex-value to write is length 
    result = tohex(len(input)) 

    # Then we write each of the elements 
    for n in input: 
     result += " " + tohex(n) 

    return result