🔍Как узнать номер столбца в pandas: простое руководство🐼
get_loc()
для объекта DataFrame. Он принимает имя столбца в качестве аргумента и возвращает индекс (номер) столбца.
Вот пример:
import pandas as pd
# Создаем DataFrame
data = {'Имя': ['Анна', 'Иван', 'Мария'],
'Возраст': [25, 30, 35],
'Город': ['Москва', 'Санкт-Петербург', 'Киев']}
df = pd.DataFrame(data)
# Получаем номер столбца
column_name = 'Город'
column_index = df.columns.get_loc(column_name)
print(f'Номер столбца "{column_name}": {column_index}')
Детальный ответ
Как узнать номер столбца в Pandas?
В библиотеке Pandas, которая широко используется для обработки и анализа данных, есть несколько способов узнать номер столбца в DataFrame. Ниже приведены два наиболее распространенных способа:
1. Использование метода get_loc()
import pandas as pd
# Создание DataFrame
data = {'Столбец1': [1, 2, 3],
'Столбец2': [4, 5, 6],
'Столбец3': [7, 8, 9]}
df = pd.DataFrame(data)
# Получение номера столбца по его имени
номер_столбца = df.columns.get_loc('Столбец2')
print(f'Номер столбца: {номер_столбца}')
В этом примере создается DataFrame с тремя столбцами. Метод get_loc() возвращает номер столбца по его имени. В данном случае номер столбца с именем 'Столбец2' будет равен 1. Обратите внимание, что отсчет номеров столбцов начинается с нуля.
2. Использование метода index()
import pandas as pd
# Создание DataFrame
data = {'Столбец1': [1, 2, 3],
'Столбец2': [4, 5, 6],
'Столбец3': [7, 8, 9]}
df = pd.DataFrame(data)
# Получение номера столбца по его имени
номер_столбца = df.columns.to_list().index('Столбец2')
print(f'Номер столбца: {номер_столбца}')
Второй способ заключается в использовании метода index(), который возвращает индекс элемента в списке столбцов DataFrame. Здесь также будет получен номер столбца с именем 'Столбец2' равный 1.
Оба этих способа позволяют быстро и удобно узнать номер столбца в Pandas DataFrame. Вы можете выбрать любой из них в зависимости от предпочтений и удобства использования.