使用pyplot绘制曲线

Question

我有2条曲线和5个数据点：

a = np.array([['16x28', 0.023392, 0.006976],
              ['64x36', 0.029888, 0.039104],
              ['512x288', 0.033328, 2.198592],
              ['1024x576', 0.065632, 5.864992],
              ['3840x2160', 0.801120, 76.550461]])

当我绘制它时，我希望有2条曲线，但会得到直线：

plt.xticks([0,1,2,3, 4],a[:,0])
plt.plot(a[:,1])
plt.plot(a[:,2])

如何使pyplot绘图曲线像它应该的那样？

更新：我使用Python 3.5.2和matplotlib 2.1.1

Answer 1

Matplotlib <2.1不允许直接绘制字符串，但是matplotlib 2.1可以。 如果检查数据，您将看到它们是字符串：

print (a[1,1], type(a[1,1]))
# 0.029888 <class 'numpy.str_'>

因此，您需要将数据转换为浮点数。 您可以使用numpy.ndarray.astype进行此操作。

import matplotlib.pyplot as plt
import numpy as np

a = np.array([['16x28', 0.023392, 0.006976],
              ['64x36', 0.029888, 0.039104],
              ['512x288', 0.033328, 2.198592],
              ['1024x576', 0.065632, 5.864992],
              ['3840x2160', 0.801120, 76.550461]])


plt.xticks([0,1,2,3,4], a[:,0])
plt.plot(a[:,1].astype(float))
plt.plot(a[:,2].astype(float))
plt.show()

赠送：