Python：对二维数组（列表的列表）中的元素执行操作的单行？

Question

我有一个列表列表，每个列表包含不同数量的字符串。我想（有效地）将这些全部转换为整数，但感觉有点密集，因为我无法让它在我的一生中发挥作用。我一直在尝试： newVals = [int(x) for x in [row for rows in values]]

其中“values”是列表的列表。它一直说 x 是一个列表，因此不能作为 int() 的参数。显然我在这里做了一些愚蠢的事情，那是什么？对于这种事情有一个公认的习语吗？

Answer 1

这使得整数嵌套起来。

[map(int, x) for x in values]

如果你想让它们扁平化，这对于 Python3 map() 返回一个迭代器来说也不难

。您可以使用

[list(map(int, x)) for x in values]

，但在这种情况下您可能更喜欢使用嵌套 LC

[[int(y) for y in x] for x in values]

Answer 2

怎么样：

>>> 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]]

Answer 3

另一种解决方法

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

Answer 4

您只需使用不正确的顺序和括号 - 应该是：

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)

Answer 5

一种丑陋的方法是使用 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]

Answer 6

为了映射具有任意数量维度的列表，您可以使用 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]]]