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

Question

Я хотел создать переменные в python на основе имен столбцов моего dataframe.Не уверен, что это возможно, так как я новичок в Python.Допустим, мой df выглядит так:

ID  Date         Name    Counts
1   04/04/2018   JUOM    150
2   05/04/2018   1XMK    22
3   05/04/2018   N2IS    130

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

Answer 1

Если вы хотите, чтобы переменная x была назначена столбцам Date и Name, вы можете установить подкадр данных с использованием синтаксиса df[['col1','col2',...]]. Внешние скобки являются операцией индексатора объекта фрейма данных pandas, а внутренние скобки создают список желаемых имен столбцов.

x= df[['Date','Name']]

Что возвращает:

         Date  Name
0  04/04/2018  JUOM
1  05/04/2018  1XMK
2  05/04/2018  N2IS

Если вы не хотите, чтобы данные были во фрейме данных, вы можете получить данные с помощью свойства .values.

x= df[['Date','Name']].values

array([['04/04/2018', 'JUOM'],
       ['05/04/2018', '1XMK'],
       ['05/04/2018', 'N2IS']], dtype=object)

Если вы используете версию 0.24.0 для панд, они рекомендуют заменить .values на to_numpy().