显示数组的第一个元素的图

Question

我有一个数组：

[[5, 6, 9,...], [3, 7, 7,...], [8, 4, 9,...],...]

如何使用matplotlib制作在y轴上显示这些数组的第一个元素的图？ x轴可以只是1,2,3 ......

因此情节将具有值：

x -> y
1 -> 5
2 -> 3
3 -> 8 ...

Answer 1

只需选择数组的第一列并使用plt.plot命令绘制它，如下所示：

import matplotlib.pylab as plt
import numpy as np

# test data
a = np.array([[5, 6, 9], [3, 7, 7], [8, 4, 9]])
print(a[:,0]) # result is [5 3 8]

# plot the line
plt.plot(a[:,0])
plt.show()

Answer 2

您可以获取列表的第一个元素，然后通过附加这些元素来创建另一个列表。

import matplotlib.pyplot as plt

oldList = [[5, 6, 9,...], [3, 7, 7,...], [8, 4, 9,...],...]
newList= []

for element in oldList:
  newList.append(element[0]) #for every element, append first member of that element

print(newList) #not necessary line, just for convenience

plt.plot(newList)
plt.show()