从其他多列中减去一列

Question

我有一个 7 列的数组，我试图从所有其他列中减去第一列。现在我的代码如下所示：

a  = (q[:,0]-q[:,1])**2
a1 = (q[:,0]-q[:,2])**2
a2 = (q[:,0]-q[:,3])**2
a3 = (q[:,0]-q[:,4])**2
a4 = (q[:,0]-q[:,5])**2
a5 = (q[:,0]-q[:,6])**2

这很好用，只是我尝试对许多文件执行此操作，而且效率极低。有没有更有效的方法来编写相同的代码？

谢谢

Answer 1

如果你想从所有其他列中减去第一列，你可以这样做

import numpy as np
x = np.arange(40).reshape(8, 5) #sample data
y = (x - x[:,0,None])**2

x[:,0,None] 代表第一列。如果您只是尝试x - x[:,0]，numpy 无法将形状一起广播（在这种情况下为 (8, 5) 和 (8,)）。添加x[:,0,None]（或x[:,0,np.newaxis]）使第二个形状（8, 1）和numpy可以一起广播两者。