熊猫数据更新行

Question

假设我有一个由0开始的数据，X和维数m。我在pandas.series，Y中有n个唯一值(1,2,3，…，n)，它的长度为m。我如何有效地设置X的第一行的I列(0到1)，而不使用循环。特别是对于大的m和n。

For example, for Y = [3,2,1]
X
row     1       2      3
0       0       0      0
1       0       0      0
2       0       0      0

to
row     1       2      3
0       0       0      1
1       0       1      0
2       1       0      0

Answer 1

我不知道你为什么反对支持循环。这应该是相当有效的。

for row, col in enumerate(Y):
    df.iat[n, col] = 1

您还可以计算索引位置并将其值设置为一个，然后将结果重塑为矩阵的m x n形状。

Y = [3, 2, 1]
n = 5
m = len(Y)
locations = set(row * n + col for row, col in enumerate(Y))
df = pd.DataFrame(
    np.array([1 if idx in locations else 0 for idx in range(m * n)]).reshape((m, n))
)
>>> df
   0  1  2  3  4
0  0  0  0  1  0
1  0  0  1  0  0
2  0  1  0  0  0