python如何替换array（x，n）中每个x的第一个f（x）值

Question

我想将数组a 每一行中的前x 值替换为1，并保留所有其他值NaN。然而，前 x 值在每一行中都会发生变化，并由列表 b 确定。

由于我对数组不是很熟悉，我想我可以使用如下所示的 for 循环来执行此操作，但这不起作用 （我从How to set first N elements of array to zero? 获得了关于数组替换基础知识的灵感）。

In:
a = np.empty((3,4))
a.fill(np.nan)
b = [2,3,1]

for i in range(b):
    a[0:b[i]] = [1] * b[i] 
    a[i:] = np.ones((b[i]))
pass

Out:
line 7:
ValueError: could not broadcast input array from shape (2) into shape (2,4)

结果应该是这样的：

Out:
[[1, 1, nan, nan], 
 [1, 1, 1, nan], 
 [1, nan, nan, nan]] 

Answer 1

在链接的答案中，How to set first N elements of array to zero?

数组的解决方案是

y = numpy.array(x)
y[0:n] = 0

换句话说，如果我们用相同的数字填充一个切片（索引范围），我们可以指定一个标量。它可以是相同大小的数组，例如np.ones(n)。但不一定是这样。

所以我们只需要遍历a 的行（以及b 的元素）并执行这个索引赋值

In [368]: a = np.ones((3,4))*np.nan
In [369]: for i in range(3):
     ...:     a[i,:b[i]] = 1
     ...:     
In [370]: a
Out[370]: 
array([[  1.,   1.,  nan,  nan],
       [  1.,   1.,   1.,  nan],
       [  1.,  nan,  nan,  nan]])

有多种方法可以用nan“填充”原始数组。 np.full 执行 np.empty 后跟 copyto。

行迭代的一个变体是for i,n in enumerate(a):。

在协调意义上进行迭代的另一种好方法是使用zip。

In [371]: for i,x in zip(b,a):
     ...:     x[:i] = 1

这利用了 2d 数组上的迭代迭代其行这一事实。所以x 是a 的一维视图，可以就地更改。

但是有了一些索引技巧，我们甚至不必循环。

In [373]: a = np.full((3,4),np.nan)

In [375]: mask = np.array(b)[:,None]>np.arange(4)
In [376]: mask
Out[376]: 
array([[ True,  True, False, False],
       [ True,  True,  True, False],
       [ True, False, False, False]], dtype=bool)
In [377]: a[mask] = 1
In [378]: a
Out[378]: 
array([[  1.,   1.,  nan,  nan],
       [  1.,   1.,   1.,  nan],
       [  1.,  nan,  nan,  nan]])

这是numpy 顶级海报@Divakar 的最爱之一。

Numpy: Fix array with rows of different lengths by filling the empty elements with zeros

它可以用于pad 一个列表列表。说到填充，itertools 有一个方便的工具，zip_longest（py3 名称）

In [380]: np.array(list(itertools.zip_longest(*[np.ones(x).tolist() for x in b],fillvalue=np.nan))).T
Out[380]: 
array([[  1.,   1.,  nan],
       [  1.,   1.,   1.],
       [  1.,  nan,  nan]])

---

您的问题应该说明出了什么问题；你遇到了什么样的错误：

for i in w2:
    a[0:b[i]] = [1] * b[i] 
    a[i:] = np.ones((b[i]))

w2 未指定，但可能是range(3)。

a[0:b[i]] 是错误的，因为它指定了所有行，而您一次只处理一个行。 a[i:] 也指定了行的范围。

Answer 2

您可以通过循环执行此操作。初始化nan 值的数组，然后循环遍历第一个 n 的列表，并根据每行的 n 将值设置为 1。

a = np.full((3, 4), np.nan)
b = [2, 3, 1]
for i, x in enumerate(b):
    a[i, :x] = 1

Answer 3

您可以使用列表推导来初始化您的矩阵：

>>> import numpy as np
>>> b = [2, 3, 1]
>>> max_len = 4
>>> gen_array = lambda i: [1] * i + [np.NAN] * (max_len - i)
>>> np.matrix([gen_array(i) for i in b])

详细步骤：

[1] * N 将创建一个长度为 N 并用 1 填充的数组：

>>> [1] * 3
[1, 1, 1]

您可以使用+ 连接数组：

>>> [1, 2] + [3, 4]
[1, 2, 3, 4]

您只需将两者结合起来 [1] * X + [np.NAN] * (N - X) 将创建一个 N 维度的数组，其中填充有 X 1

最后一个，列表理解：

[i for i in b]

是一个“捷径”（不是真的，但更容易理解）：

a = []
for i in b:
    a.append(i)

Answer 4

import numpy as np
a = np.random.rand(3,4)  #Create matrix with random numbers (you can change this to np.empty or whatever you want.
b = [1, 2, 3] # Your 'b' list
for idr, row in enumerate(a): # Loop through the matrix by row
  a[idr,:b[idr]] = 1  # idr is the row index, here you change the row 'idr' from the column 0 to the column b[idr] that will be 0, 1 and 3
  a[idr,b[idr]:] = 'NaN'  # the values after that receive NaN
print(a) # Outputs matrix
#[[  1.  nan  nan  nan]
 [  1.   1.  nan  nan]
 [  1.   1.   1.  nan]]