Question

我想知道是否可以在不使用for循环的情况下将函数（或juste操作，例如替换值）应用于python 2d数组中的列。

如果已经提出问题，我很抱歉，但我找不到任何有关我问题的具体信息。

我想做点什么：

array[:][2] = 1

这意味着为第三列的每个值设置1，或

func(array[:][2])

这意味着将func()应用于数组的第三列。

有没有神奇的蟒蛇方法呢？

编辑：事实已经说过了。我忘了说我不想避免for()语句来提高性能，只是因为我不想为这个精确的实例添加多行。我们在这里得到了2个答案，一个是本地方式，另外两个是在Numpy的帮助下。非常感谢您的回答！

Answer 1

您可以使用numpy数组轻松完成此操作。示例 -

In [2]: import numpy as np

In [3]: na = np.array([[1,2,3],[3,4,5]])

In [4]: na
Out[4]:
array([[1, 2, 3],
       [3, 4, 5]])

In [5]: na[:,2] = 10

In [6]: na
Out[6]:
array([[ 1,  2, 10],
       [ 3,  4, 10]])

In [7]: na[:,2]
Out[7]: array([10, 10])

In [8]: def func(a):
   ...:     for i,x in enumerate(a):
   ...:         a[i] = x + 1
   ...:

In [9]: na
Out[9]:
array([[ 1,  2, 10],
       [ 3,  4, 10]])

In [10]: func(na[:,1])

In [11]: na
Out[11]:
array([[ 1,  3, 10],
       [ 3,  5, 10]])

您可以找到有关此here的更多详细信息。对于numpy数组，请注意in documentation -

基本切片生成的所有数组始终是原始数组的views。

这就是为什么在更改函数内部的切片数组时，实际的数组已经改变了。

Answer 2

没有numpy可以这样做：

map(lambda x: x[:2] + [1] + x[3:], array)

map(lambda x: x[:2] + my_func(x[2]) + x[3:], array)

Answer 3

在numpy中非常简单，你可以通过一个简单的任务来完成它：

>>> numpy.array[:,column_number]=value

但如果您正在寻找python方法，可以使用zip函数和itertools.repeat()：

>>> from itertools import repeat
>>> def replacer(l,index,value):
...     z=zip(*l)
...     z[index]=list(repeat(value,len(l)))
...     return zip(*z)

演示：

>>> l=[range(4) for _ in range(3)]
>>> replacer(l,2,'*')
[(0, 1, '*', 3), (0, 1, '*', 3), (0, 1, '*', 3)]

注意，因为在python 3.X zip返回一个迭代器，你可以使用list函数返回一个列表，因为迭代器不支持你需要调用列表的函数内的索引

>>> def replacer(l,index,value):
...     z=list(zip(*l))
...     z[index]=list(repeat(value,len(l)))
...     return zip(*z)

>>> list(replacer(l,2,'*'))
[(0, 1, '*', 3), (0, 1, '*', 3), (0, 1, '*', 3)]

Answer 4

你可以将一个函数映射到每一列......

ar = [[1,2,3],
      [3,4,5],
      [3,4,5],
      [3,4,5],
      [3,4,5]]

if you want change second column to 100:

    def column_change(l,n =1 , m =100):
        l[n] = m
        return l

   print(list(map(colum_change,ar)))

    [[1, 100, 3], [3, 100, 5], [3, 100, 5], [3, 100, 5], [3, 100, 5]]

在2d阵列列上操作

4 个答案: