Как установить имена столбцов в pandas: полезные советы и трюки 🐼

    import pandas as pd

    # Создание DataFrame с названиями столбцов
    df = pd.DataFrame(data=[[1, 2, 3], [4, 5, 6]], columns=['Столбец 1', 'Столбец 2', 'Столбец 3'])
    

Как задать имена столбцов в Pandas

В библиотеке Pandas, допустимо задавать пользовательские имена столбцов для удобства работы с данными. В этой статье я расскажу вам о различных методах, которые можно использовать для задания имен столбцов в Pandas.

1. Использование параметра 'columns' при чтении данных

При чтении данных из файла можно сразу задать имена столбцов, используя параметр 'columns' функции чтения данных, такой как read_csv(), read_excel() и другие. Ниже приведен пример:

import pandas as pd

data = pd.read_csv('data.csv', columns=['Имя', 'Возраст', 'Город'])

В этом примере мы указываем, что в файле 'data.csv' столбцы должны называться 'Имя', 'Возраст' и 'Город'.

2. Использование метода 'rename()' для переименования столбцов

Другим способом задания имен столбцов является использование метода rename(). Этот метод позволяет переименовывать столбцы по указанному словарю со старыми и новыми именами столбцов. Вот пример:

import pandas as pd

data = pd.read_csv('data.csv')
data = data.rename(columns={'old_name1': 'new_name1', 'old_name2': 'new_name2'})

В этом примере мы читаем данные из файла 'data.csv' и затем с помощью метода rename() переименовываем столбцы 'old_name1' и 'old_name2' в 'new_name1' и 'new_name2' соответственно.

3. Использование атрибута 'columns' для присваивания новых имен столбцам

Третий способ задания имен столбцов в Pandas - использование атрибута 'columns' для непосредственного присваивания списка новых имен столбцам. Вот пример:

import pandas as pd

data = pd.read_csv('data.csv')
data.columns = ['new_name1', 'new_name2', 'new_name3']

В этом примере мы сначала читаем данные из файла 'data.csv', а затем присваиваем новые имена столбцам, используя атрибут 'columns' и список новых имен.

4. Использование метода 'set_axis()' для задания имен столбцам

Четвертый способ задания имен столбцов - использование метода set_axis(). Этот метод позволяет задать имена столбцов путем передачи списка новых имен и оси, к которой применяются имена. Пример использования:

import pandas as pd

data = pd.read_csv('data.csv', header=None)
data.set_axis(['new_name1', 'new_name2', 'new_name3'], axis=1, inplace=True)

В этом примере мы сначала читаем данные из файла 'data.csv', используя параметр 'header=None', чтобы не использовать первую строку файла в качестве имен столбцов. Затем мы используем метод set_axis(), чтобы задать новые имена столбцам.

Заключение

Существует несколько способов задания имен столбцов в Pandas, как при чтении данных, так и после. В этой статье мы обсудили использование параметра 'columns' при чтении данных, метода 'rename()', атрибута 'columns' и метода 'set_axis()'. Вы можете выбрать подходящий способ в зависимости от вашего конкретного случая использования.

Удачи в работе с Pandas!