从.csv文件生成图

时间:2016-10-13 23:49:07

标签: python csv matplotlib

我试图从NASA的Exoplanet Archive制作几张Exoplanet数据图。问题是我没做什么会返回csv文件的列,其中包含csv文件第一行的标题。

我得到的错误是

    NameError: name 'pl_orbper' is not defined

https://groups.google.com/forum/#!topic/nosql-databases/2vBHDhGti9E

我目前没有使用的代码虽然我确定我已经关闭了。

   import matplotlib.pyplot as plt
   import numpy as np

   data = np.genfromtxt("planets.csv",delimiter=',',names=True, unpack=True)

   plt.plot(pl_orbper,pl_bmassj)
   plt.title('Mass vs Period')
   plt.ylabel('Period')
   plt.xlabel('Mass')

   plt.show() 

如果有人使用csv.reader或任何其他方式阅读csv文件有更好的解决方案,我会对它开放。

1 个答案:

答案 0 :(得分:0)

更改以下行:

plt.plot(pl_orbper,pl_bmassj)

plt.plot(data['pl_orbper'],data['pl_bmassj'])

使用以下数据:

rowid,pl_orbper,pl_bmassj
1, 326.03, 0.320
2, 327.03, 0.420
3, 328.03, 0.520
4, 329.03, 0.620
5, 330.03, 0.720
6, 331.03, 0.820

这是结果

Mass vs Period Plot