Python 自动设置图像标签

我有一个2D数组，我在一个图像中绘制成三行，每行代表每列中的数字

我想根据线条所代表的列，自动将它们的标签设置为“column=1”、“column=2”、“column=3”

我怎样才能做到这一点

#%%
import numpy as np
import matplotlib.pyplot as plt

data = [
    [1, 2, 4],
    [2, 4, 6],
    [5, 7, 9]
]

arr = np.array(data)

#Plotting

arr = arr[:,0:3]
fig, ax = plt.subplots(figsize=(8, 5))

x = np.linspace(0, 2, 3)
x = np.asarray(x)

plt.plot(x, arr, label="column")
plt.legend(bbox_to_anchor=(0.85, 1), loc='upper left', borderaxespad=0.)
plt.show()

---

您可以在

图例

方法中设置标签

...
lines = plt.plot(x, arr)
column_names = tuple("column_" + str(i+1) for i,x in enumerate(data))
plt.legend(lines, column_names, bbox_to_anchor=(0.85, 1), loc='upper left', borderaxespad=0.)
plt.show()