在python上绘制3D数据

Question

我是python编程的新手，我正在开发一个应用程序，该应用程序可以从3个列中绘制3D表面或轮廓图，这些列是从文本文件中获取的，并以逗号分隔。 我从使用地理坐标的3D图中获得了提示，但我不断收到IndexError：列表索引超出范围错误。 这是我的代码：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import sys

ID = []
lat_col = []
long_col = []
Bouguer_col = []
cluster1 = 'Data.txt'
with open(cluster1) as f:
    lines = f.readlines()
    for line in lines:
        items = line.strip().split()
        lat = float(items[1])
        lon = float(items[2])
        bga = float(items[3])
        Bouguer_Anomaly = float(items[4])
        lat_col.append(lat)
        long_col.append(lon)
        Bouguer_col.append(bga)
        ID.append(Bouguer_Anomaly)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
p = ax.scatter(long_col, lat_col, Bouguer_col, c=ID, marker='o')

ax.set_xlabel('Longitude')
ax.set_ylabel('Latitude')
ax.set_zlabel('Bouguer Anomaly')
ax.invert__zaxis()
cb = fig.colorbar(p, label='Difference')
plt.savefig('plot.png')
plot.show()

如果有人可以帮助我提供从电子表格的各个列中绘制x，y，z数据的代码，我也将感到高兴！

.txt文件的部分内容如下所示：

Lat, Long, Elev, ObsGrav, Anomalies
6.671482000000001022e+00,7.372505999999999560e+00,3.612977999999999952e+02,9.780274000000000233e+05,-1.484474523360840976e+02
6.093078000000000216e+00,7.480882000000001142e+00,1.599972999999999956e+02,9.780334000000000233e+05,-1.492942383352201432e+02
6.092045999999999850e+00,7.278669999999999973e+00,1.462445999999999913e+02,9.780663000000000466e+05,-1.190960417173337191e+02
6.402087429999999912e+00,7.393360939999999992e+00,5.237939999999999827e+02,9.780468000000000466e+05,-8.033459449396468699e+01
6.264082730000000154e+00,7.518244540000000420e+00,2.990849999999999795e+02,9.780529000000000233e+05,-1.114865156192099676e+02
6.092975000000000030e+00,7.482914000000000065e+00,1.416474000000000046e+02,9.780338000000000466e+05,-1.525697779102483764e+02
6.383570999999999884e+00,7.289616999999999791e+00,2.590403000000000020e+02,9.780963000000000466e+05,-8.300666170357726514e+01
6.318417000000000172e+00,7.557638000000000744e+00,1.672036999999999978e+02,9.780693000000000466e+05,-1.246774551668204367e+02
6.253779999999999895e+00,7.268805999999999656e+00,1.059429999999999978e+02,9.781026999999999534e+05,-9.986763240839354694e+01
6.384635000000000282e+00,7.291032000000000401e+00,2.615624000000000251e+02,9.780963000000000466e+05,-8.256190758384764194e+01

我需要在3D表面或彩色轮廓图上绘制[纬度，经度和异常]。

Answer 1

当您观察到.csv文件具有标头时，因此无法将其转换为浮动标头，适当的做法是跳过第一行，然后还有其他印刷错误，我已在以下代码中纠正了这些错误：

from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import sys

ID = []
lat_col = []
long_col = []
Bouguer_col = []
cluster1 = 'data.txt'
with open(cluster1) as f:
    lines = f.readlines()
    for line in lines[1:]:
        lat, lon, elev, bga, Bouguer_Anomaly = [float(i) for i in line.strip().split(",")]
        lat_col.append(lat)
        long_col.append(lon)
        Bouguer_col.append(bga)
        ID.append(Bouguer_Anomaly)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
p = ax.scatter(long_col, lat_col, Bouguer_col, c=ID, marker='o')

ax.set_xlabel('Longitude')
ax.set_ylabel('Latitude')
ax.set_zlabel('Bouguer Anomaly')
ax.invert_zaxis()
cb = fig.colorbar(p, label='Difference')
plt.savefig('plot.png')
plt.show()