Как добавить строку заголовка в Pandas DataFrame(с примерами)

Редакция Кодкампа -

Вы можете использовать один из следующих трех методов, чтобы добавить строку заголовка в кадр данных pandas:

#add header row when creating DataFrame
df = pd.DataFrame(data=[data_values],
 columns=['col1', 'col2', 'col3'])

#add header row after creating DataFrame
df = pd.DataFrame(data=[data_values])
df.columns = ['A', 'B', 'C']

#add header row when importing CSV
df = pd.read_csv('data.csv', names=['A', 'B', 'C'])

В следующих примерах показано, как использовать каждый из этих методов на практике.

Пример 1: добавить строку заголовка при создании DataFrame

В следующем коде показано, как добавить строку заголовка при создании кадра данных pandas:

import pandas as pd
import numpy as np

#add header row when creating DataFrame 
df = pd.DataFrame(data=np.random.randint (0, 100, (10, 3)),
 columns =['A', 'B', 'C'])

#view DataFrame
df

 A B C
0 81 47 82
1 92 71 88
2 61 79 96
3 56 22 68
4 64 66 41
5 98 49 83
6 70 94 11
7 1 6 11
8 55 87 39
9 15 58 67

Пример 2: добавить строку заголовка после создания DataFrame

В следующем коде показано, как добавить строку заголовка после создания кадра данных pandas:

import pandas as pd
import numpy as np

#create DataFrame
df = pd.DataFrame(data=np.random.randint (0, 100, (10, 3))) 

#add header row to DataFrame
df.columns = ['A', 'B', 'C']

#view DataFrame
df

 A B C
0 81 47 82
1 92 71 88
2 61 79 96
3 56 22 68
4 64 66 41
5 98 49 83
6 70 94 11
7 1 6 11
8 55 87 39
9 15 58 67

Пример 3: добавить строку заголовка при импорте DataFrame

В следующем коде показано, как добавить строку заголовка с помощью аргумента имен при импорте кадра данных pandas из файла CSV:

import pandas as pd
import numpy as np

#import CSV file and specify header row names
df = pd.read_csv('data.csv', names=['A', 'B', 'C'])

#view DataFrame
df

 A B C
0 81 47 82
1 92 71 88
2 61 79 96
3 56 22 68
4 64 66 41
5 98 49 83
6 70 94 11
7 1 6 11
8 55 87 39
9 15 58 67

Связанный: Как читать файлы CSV с помощью Pandas

Дополнительные ресурсы

Как добавить строки в Pandas DataFrame
Как добавить массив Numpy в фрейм данных Pandas
Как подсчитать количество строк в Pandas DataFrame