python - 如何在绘图之前将模块转换为列表或数组？

Question

我在 PyCharm 中有 2 个文件：

比喻.py：

def function_parable(a):

    return [x**2 for x in a]

---

主要.py：

import numpy as np
import matplotlib.pyplot as plt
a = [1, 2, 3, 4, 5]
a = np.array(a)
n = a.shape
print(n)
print(type(a))
print(a)

import parable as b
print(b.function_parable(a))
print(type(b))


plt.plot(a,b)
plt.show()

---

错误：ValueError：x 和 y 必须具有相同的第一维，但具有形状 (5,) 和 (1,)

我该如何维护这个结构并解决这个问题？谢谢！

标签： pythonfunctionnumpymatplotlibplot

Answer 1

尝试以下脚本：

import numpy as np
import matplotlib.pyplot as plt
a = [1, 2, 3, 4, 5]
a = np.array(a)
n = a.shape
print(n)
print(type(a))
print(a)

import parable as f
b = f.function_parable(a)
print(b)
print(type(b))


plt.plot(a,b)
plt.show()

快乐编码我的朋友。