[FIXED] Plotting 3D data on python

Issue

I'm new to python programming and I'm working on an app that can plot 3D surface or contour graph from 3 columns obtained from a text file with comma-separated values. I got hints from 3D plot using geographic coordinates but I keep getting IndexError: list index out of range error. This is my code:

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()

I'll also be glad if anyone can help me with the code for plotting x,y,z data from individual columns of a spreadsheet!

Part of the content of the .txt file is shown below:

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

I need to plot [Lat, Long and Anomalies] on a 3D surface or a coloured contour graph.

Solution

As you observe your .csv file has a header, so you can not convert it to floating, the appropriate thing is to skip the first line, then you have other typographical errors that I have corrected in the following code:

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()

Answered By - eyllanesc