使用元组跳过列填充numpy.darray

时间:2017-03-07 13:31:20

标签: python numpy

我有一个使用np.zeros([m, n])的m x n数组,我想用返回的元组填充一些行(例如第0行)。但是我想跳过某些应保持为0的列。

现在我必须重复该功能(或将它们存储在某处)并填充该行的某些部分。

使用返回长度为6的元组的函数的示例

A[0,0:2] = someClass.someFunc(var1, var2)[0:2]
A[0,4:8] = someClass.someFunc(var1, var2)[2:6]

我用元组的前2个变量填充前2列,跳过2行,然后用元组的剩余部分填充以下4列。

有没有办法实现这样的目标:

A[0,0:2], A[0,4:8] = someClass.someFunc(var1, var2)

是否需要重复此功能?

1 个答案:

答案 0 :(得分:4)

您可以使用np.r_连接这些范围以简化左侧 -

A[0,np.r_[0:2,4:8]] = someClass.someFunc(var1, var2)