10 
# 2x3 dimensional list
multidim_list = [ 
                  [1,2,3],
                  [4,5,6],    
                ]
# 2x3x2 dimensional list
multidim_list2 = [ 
                   [ 
                     [1,2,3],
                     [4,5,6],  
                   ],
                   [ 
                     [7,8,9],
                     [10,11,12],  
                   ]
                 ]

def multiply_list(list):
    ...

我想实现一个函数，它将列表中的所有元素乘以 2。但是我的问题是列表可以有不同数量的维度。

有没有一种通用的方法来循环/迭代多维列表，例如将每个值乘以 2？

EDIT1：感谢您的快速回答。对于这种情况，我不想使用 numpy. 递归看起来不错，甚至不需要复制列表，实际上可能非常大。

4

3 回答 3

15

递归是你的朋友：

from collections import MutableSequence
def multiply(list_):
    for index, item in enumerate(list_):
        if isinstance(item, MutableSequence):
            multiply(item)
        else:
            list_[index] *= 2

您可以只做isinstance(item, list)而不是isinstance(item, MutableSequence)，但后一种方式更具前瞻性和通用性。有关简短说明，请参阅词汇表。

于 2012-04-22T20:54:15.703 回答
8

您可以使用 numpy：

import numpy as np

arr_1 = np.array(multidim_list)
arr_2 = np.array(multidim_list2)

结果：

>>> arr_1*2
array([[ 2,  4,  6],
       [ 8, 10, 12]])
>>> arr_2*2
array([[[ 2,  4,  6],
        [ 8, 10, 12]],

       [[14, 16, 18],
        [20, 22, 24]]])
于 2012-04-22T20:54:26.837 回答
3

numpy数组开箱即用。

于 2012-04-22T20:54:20.483 回答