我有一个列表列表,每个列表都包含不同数量的字符串。我想(有效地)将这些全部转换为整数,但我感觉有点密集,因为我无法让它为我的生活解决问题。我一直在尝试: newVals = [int(x) for x in [row for rows in values]]
其中“值”是列表列表。它一直说 x 是一个列表,因此不能作为 int() 的参数。显然我在这里做了一些愚蠢的事情,它是什么?这种事情有公认的成语吗?
这使整数嵌套
[map(int, x) for x in values]
如果你想让它们变平,那也不难
对于 Python3map()
返回一个迭代器。你可以使用
[list(map(int, x)) for x in values]
但在这种情况下,您可能更喜欢使用嵌套的 LC
[[int(y) for y in x] for x in values]
怎么样:
>>> a = [['1','2','3'],['4','5','6'],['7','8','9']]
>>> [[int(j) for j in i] for i in a]
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
您只需使用不正确的顺序和括号 - 应该是:
inputVals = [['1','2','3'], ['3','3','2','2']]
[int(x) for row in inputVals for x in row]
或者,如果您需要输出列表中的列表,则:
map(lambda row: map(int, row), inputVals)
Another workaround
a = [[1, 2, 3], [7, 8, 6]]
list(map(lambda i: list(map(lambda j: j - 1, i)), a))
[[0, 1, 2], [6, 7, 5]] #output
一个丑陋的方法是使用 evalf:
>>> eval(str(a).replace("'",""))
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
如果你不介意一个数组中的所有数字,你可以去:
>>> a = [['1','2','3'],['4','5','6'],['7','8','9']]
>>> map(int,sum(a,[]))
[1, 2, 3, 4, 5, 6, 7, 8, 9]
为了映射具有任意数量维度的列表,您可以使用 numpy.apply_over_axes
import numpy as np
np.apply_over_axes(lambda x,_:x*2, np.array([[1,2,3],[5,2,1]]),[0])
--------------------
array([[ 2, 4, 6],
[10, 4, 2]])
不幸的是,如果您还需要更改变量类型,这将不起作用。没有找到任何库解决方案,所以这里是执行此操作的代码:
def map_multi_dimensional_list(l, transform):
if type(l) == list and len(l) > 0:
if type(l[0]) != list:
return [transform(v) for v in l]
else:
return [map_multi_dimensional_list(v, transform) for v in l]
else:
return []
map_multi_dimensional_list([[[1,2,3],[5,2,1]],[[10,20,30],[50,20,10]]], lambda x:x*2)
------------------
[[[2, 4, 6], [10, 4, 2]], [[20, 40, 60], [100, 40, 20]]]