Установка меток для изображения автоматически

Question

У меня есть двумерный массив, и я делю график на три строки на одном изображении, каждая строка представляет числа в каждом столбце.

Я бы хотел установить их метки автоматически как «столбец = 1», «столбец = 2», «столбец = 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()

Answer 1

Вы можете установить этикетки в методе legend,

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