Question

我有这个m x n numpy数组，我想对行元素应用某些操作。虽然，只能将其强制转换为索引在索引向量上的条目之前指定的那些元素。

我已经经历了经典的for循环方法，但是我期待更多NumPythonic。

以下代码将完成工作：

for i,j in enumerate(x):
    M[i, 0:j] = 2*M[i, 0:j]

但是我一直在寻找一种广播的，不可取的方法。有任何想法吗？

例如，可以说

M = [[ 1, 2, 3, 4, 5],
     [ 6, 7, 8, 9,10]]

x = [2, 3]

，我们的应用是将某些元素加倍。根据x中指定的索引，我们应该具有结果数组：

M = [[ 2, 4, 3, 4, 5],
     [12,14,16, 9,10]]

Answer 1

以下是两种相关的方法，它们的速度大致相同：

import numpy as np
from timeit import timeit

M = [[ 1, 2, 3, 4, 5],
     [ 6, 7, 8, 9,10]]

x = [2, 3]

def f():
    MM = np.array(M)
    xx = np.array(x)
    MM[np.arange(5)<xx[:,None]] *= 2
    return MM

def g():
    MM = np.array(M)
    xx = np.array(x)
    MM *= 1 + (np.arange(5)<xx[:,None])
    return MM

print(f())
print(g())
M = 1000*M
x = 1000*x
print(timeit(f,number=1000))
print(timeit(g,number=1000))

样品运行：

[[ 2  4  3  4  5]
 [12 14 16  9 10]]
[[ 2  4  3  4  5]
 [12 14 16  9 10]]
1.1994759310036898
1.1547658089984907

如何通过切片索引向量来修改np数组？

1 个答案: