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

Question

我在 PyCharm 中有 2 个文件:
parable.py:

def function_parable(a):

    return [x**2 for x in a]

main.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,)
我该如何维护这个结构并解决这个问题？谢谢!

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()

快乐编码我的 friend 。