Получить имя столбца из фрейма данных в виде строки

Question

Как я могу взять имя столбца в фрейме данных из списка и добавить его в текст?

column=[df['o2_dry'],df['co2_dry'],df['co_ppm_dry'],df['no_ppm_dry'],df['velocity-magnitude']]


for i in range(5):
    axs[i].title.set_text('Graph for '+ str(column[i]))

Результат должен быть:

Graph for o2_dry

Graph for co2_dry 

Et c.

Answer 1

Поскольку df[name] возвращает Series, вы можете получить как значения, так и его имя (используя .name)

columns = [df['o2_dry'],df['co2_dry'],df['co_ppm_dry'],df['no_ppm_dry'],df['velocity-magnitude']]
    
for i in range(5):
    axs[i].title.set_text('Graph for '+ column[i].name)

Но лучше было бы

columns = ['o2_dry', 'co2_dry', 'co_ppm_dry', 'no_ppm_dry', 'velocity-magnitude']
    
for idx, column in enumerate(columns):
    axs[idx].title.set_text('Graph for '+ column)
    values = df[column]

Answer 2

Во-первых, после чтения фрейма данных вы можете получить имена столбцов с помощью variable_name.columns.tolist (), где имя varibale - это имя, присвоенное вашему фрейму данных. например:

        df = pd.read_csv(path_to_file)
        column_name_list = df.columns.tolist()

тогда список имен столбцов будет содержать все имена столбцов, только l oop поверх этого списка и напечатать.

Answer 3

Вы можете использовать columns[i].name, чтобы получить имя. Обратите внимание, что columns - это список объектов pandas .series.