如何使用 Matplotlib 绘制复数（Argand 图）？

要使用 Matplotlib 绘制复数，我们可以创建一个含有复数的数据集。

步骤

• 设置图形大小并调整子图之间和周围的填充。
• 创建随机复数。
• 创建一个包含子图的图形，使用 subplots() 方法。
• 使用 scatter() 方法绘制散点。
• 使用 show() 方法显示图形。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
data = np.random.rand(10) + 1j*np.random.rand(10)
fig, ax = plt.subplots()
ax.scatter(data.real, data.imag, c=data.real, cmap="RdYlBu_r")
plt.show()

输出