Выбор столбцов

Введение в Data Science на Python

Hillary Green-Lerman

Lead Data Scientist, Looker

Зачем выбирать столбцы?

Использование в вычислениях
```
credit_records.price.sum()
```

Построение графиков

plt.plot(ransom['letter'], ransom['frequency'])

Имена столбцов — это строки

print(credit_records.head())

            suspect         location              date         item  price
0    Kirstine Smith   Groceries R Us   January 6, 2018     broccoli   1.25
1      Gertrude Cox  Petroleum Plaza   January 6, 2018  fizzy drink   1.90
2  Fred Frequentist   Groceries R Us   January 6, 2018     broccoli   1.25
3      Gertrude Cox   Groceries R Us  January 12, 2018     broccoli   1.25
4    Kirstine Smith    Clothing Club   January 9, 2018        shirt  14.25

'suspect'
'location'
'date'
'item'
'price'

Выбор через скобки и строку

suspect = credit_records['suspect']

print(suspect)

0            Kirstine Smith
1              Gertrude Cox
2          Fred Frequentist
3              Gertrude Cox
4            Kirstine Smith
5              Gertrude Cox
...
99             Gertrude Cox
100        Fred Frequentist
101            Gertrude Cox
102          Kirstine Smith
103    Ronald Aylmer Fisher

Выбор через точку

price = credit_records.price

print(price)

0       1.25
1       1.90
2       1.25
3       1.25
4      14.25
5       3.95
...
99     14.25
100    12.05
101    20.15
102     3.95
103     2.05

Частые ошибки при выборе столбцов

Если имя столбца содержит пробелы или спецсимволы (-, ? и т. д.), используйте скобки со строкой

police_report['Is Golden Retriever?']

НЕ

police_report.Is Golden Retriever?

Object `Retriever` not found.

Частые ошибки при выборе столбцов

При использовании скобок и строки не забудьте кавычки вокруг имени столбца!

credit_report['location']

НЕ

credit_report[location]

Object `location` not found.

Частые ошибки при выборе столбцов

Квадратные скобки, не круглые

credit_report['location']

НЕ

credit_report('location')

----------------------------------------------------------------------
TypeError  Traceback (most recent call last)
<ipython-input-5-aabdb8981438> in <module>()
----> 1 credit_report('location')

TypeError: 'DataFrame' object is not callable

Давайте потренируемся!

Введение в Data Science на Python