如何在python中使用几个项目向Argand Diagram添加图例？

Question

我有一个复杂数字的数组，如：

a = np.array([0.074-0.046j], [-0.005+0.091j], [0.039-0.023j], [-0.245+0.015j], [0.642-0.766j])

使用功能＆＃34; polarplot＆＃34;我可以绘制数组＆＃34; a＆＃34;：

def polarplot(a):
import matplotlib.pyplot as plt
import numpy as np
for x in a:
    plt.polar([np.angle(x)],[np.abs(x)],marker='o')
plt.show()

然后我得到了这个结果： Argand Plot

当我尝试使用plt.legend(loc="lower right")添加图例时，不会显示任何内容。如何在此情节中添加图例？

Answer 1

致电label = ''

时，您需要添加plt.polar()

因此，您的代码应如下所示：

import matplotlib.pyplot as plt
import numpy as np

a = np.array([[0.074-0.046j], [-0.005+0.091j], [0.039-0.023j], [-0.245+0.015j], [0.642-0.766j]])
def polarplot(a):    
    for x in a:
        plt.polar([np.angle(x)],[np.abs(x)],marker='o',label = '%s' %x)
    plt.legend(loc = "lower right")
    plt.show()

polarplot(a)

这给出了如下所示的图像：

请注意，您可能希望移动图例并在实际实现时更改标签。