我正在尝试在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中绘制的向量?

最佳答案

np.append与普通数组追加不同。
尝试:

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)

关于python - 在python中的数组中创建数组,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/42013384/

10-10 13:31