如何使用matplotlib.pyplot一次绘制列表？

Question

我创建了一个计算素数的程序。 这只是一个测试，但我认为将其绘制在图形上会很有趣。

import matplotlib.pyplot as plt

max = int(input("What is your max no?: "))
primeList = []

for x in range(2, max + 1):
  isPrime = True
  for y in range (2 , int(x ** 0.5) + 1):
     if x % y == 0:
        isPrime = False
        break

     if isPrime:
        primeList.append(x)


print(primeList)

然后如何绘制primeList ，我可以在for循环之外一次完成所有操作吗？

Answer 1

这将绘制您的列表及其索引：

from matplotlib import pyplot as plt

plt.plot(primeList, 'o')
plt.show()

该程序：

from matplotlib import pyplot as plt

max_= 100
primeList = []

for x in range(2, max_ + 1):
    isPrime = True
    for y in range (2, int(x ** 0.5) + 1):
        if x % y == 0:
            isPrime = False
            break
    if isPrime:
        primeList.append(x)

plt.plot(primeList, 'o')
plt.show()

产生此图：