在python中从文件读取Meshgrid数据

Question

我有以下数据文件“ data.dat”：

[[0.2 0.3 0.4][0.3 0.2 0.4][0.5 0.3 0.4]]

现在我从文件中读取此数据为

f=open("data.dat","r")
z=f.read()
f.close()

我有x=[1 2 3]和y=[1 2 3] 。 我用x和y制作了一个网格

X,Y=plt.meshgrid(x,y)

现在我正在尝试使用绘制轮廓图

plt.contourf(X,Y,Z)

但是它显示错误为：ValueError：无法将字符串转换为float：[[0.2 0.3 0.4] [0.3 0.2 0.4] [0.5 0.3 0.4]]

关于如何从文件中读取Z数组为float或以其他方式写入“ data.dat”文件的任何建议？

Answer 1

您正在从文件中读取字符串，并将该字符串放入Z 但实际上，您需要浮点数的二维数组而不是字符串。 因此，您必须解析您的字符串。 可以通过将其转换为JSON来完成，如下所示：

import json
with open("data.dat") as f:
    z = f.read()
z = z.replace(' ', ', ').replace('][', '], [')
Z = json.loads(z)

另一种（更好的方法）是使用JSON将数据存储在文件中。

import json    
Z = [[0.2, 0.3, 0.4], [0.3, 0.2, 0.4], [0.5, 0.3, 0.4]]
with open("data.dat", 'w') as f:
    json.dump(Z, f)

您还可以尝试其他格式，例如CSV。