Получение столбца «номер» и имени столбца, чтобы упростить выбор нескольких - не всегда смежных - столбцов в большом df в пандах - PullRequest
1 голос
/ 23 октября 2019

У меня большой df, из которого я хотел бы делать вычисления и прогнозы, проблема в том, что я не могу найти способ получить список всех имен столбцов ВМЕСТЕ со столбцом «число». Невозможно сосчитать сверху, чтобы увидеть, какое это число, и я бы не стал выписывать все имена столбцов.
Было бы неплохо иметь возможность использовать что-то вроде этого:
df.iloc[:, np.r_[2, 5:10, 22:102, 109:129]]
но чтобы это работало, мне нужно знать, какой столбец имеет какое число.
list(df) дает мне хороший список, но без чисел, что делает его бессмысленным в этом квесте.

1 Ответ

2 голосов
/ 23 октября 2019

Я предлагаю создать словарь с enumerate:

df = pd.DataFrame({
        'A':list('abcdef'),
         'B':[4,5,4,5,5,4],
         'C':[7,8.0,9,4.0,2,3],
         'D':[1,3,5,7,1,0],
         'E':[5,3,6,9,2,4],
         'F':list('aaabbb')
})

d = dict(enumerate(df))
print (d)
{0: 'A', 1: 'B', 2: 'C', 3: 'D', 4: 'E', 5: 'F'}

Или списком кортежей, подобным предложенному @Chris в комментариях:

L = list(enumerate(df))
print (L)
[(0, 'A'), (1, 'B'), (2, 'C'), (3, 'D'), (4, 'E'), (5, 'F')]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...