python - 在python中创建数组内的数组

Question

我正在尝试在 python 中创建一个数组中的数组，如下所示:

[[3,0,-3,-4],[6,0,-2.44,-4]]

我可以在 matplotlib 的矢量图中使用它。

我尝试使用以下程序执行此操作:

data = np.loadtxt(sys.argv[1], dtype='str',delimiter=',', skiprows=1, usecols=(0,1,2,3,4,5,6,7))

x = data[:,0].astype(float)
u = data[:,6].astype(float)
v = data[:,7].astype(float)

soa = []
for t in range(0,2):
    print "At time ",x[t]," U is ",u[t]," and V is ",v[t]
    result = [x[t],0,u[t],v[t]] 
    soa = np.append(soa, [result])


print "soa is ",soa

当我运行程序时，我得到输出:

soa 是 [ 3.0.-3. -4。 6.0. -2.44 -4。 ]

这不能在 matplotlib 中绘制为矢量图。如何调整上面的脚本以将数组转换为以下格式:

 [[3,0,-3,-4],[6,0,-2.44,-4]]

其中 [3,0,-3,-4] 和 [6,0,-2.44,-4] 是我可以在 matplotlib 中绘制的向量？

Answer 1

np.append 与 noraml 数组追加不同。

尝试:

for t in range(0,2):
    print "At time ",x[t]," U is ",u[t]," and V is ",v[t]
    result = [x[t],0,u[t],v[t]] 
    soa.append(result)