CODEWITHSUNDEEP
pythonpandasindexingboolean
FortuneFaded
FortuneFaded

Reputation: 1329

Pandas: Why are double brackets needed to select column after boolean indexing

For a df table like below,

   A B C D
0  0 1 1 1
1  2 3 5 7
3  3 1 2 8

why are the double brackets needed for selecting specific columns after boolean indexing? 

the [['A','C']] part of

df[df['A'] < 3][['A','C']]

Upvotes: 55

Views: 61370

Answers (4)

EdChum
EdChum

Reputation: 393973

Because you have no columns named 'A','C', which is what you'd be trying to do which will raise a KeyError, so you have to use an iterable to sub-select from the df.

So

df[df['A'] < 3]['A','C']

raises

KeyError: ('A', 'C')

Which is different to

>>> df[df['A'] < 3][['A','C']]
   A  C
0  0  1
1  2  5

This is no different to trying:

df['A','C']

hence why you need double square brackets:

df[['A','C']]

Note that the modern way is to use .loc:

>>> df.loc[df['A'] < 3,['A','C']]
   A  C
0  0  1
1  2  5

So that you're operating on a view rather than potentially a copy

Upvotes: 14

lxstmnk
lxstmnk

Reputation: 31

Adding to previous responses, you could also use df.iloc accessor if you need to select index positions. It's also making the code more reproducible, which is nice.

Upvotes: 3

Pavel Savara
Pavel Savara

Reputation: 3457

Because inner brackets are just python syntax (literal) for list.

The outer brackets are the indexer operation of pandas dataframe object.

In this use case inner ['A', 'B'] defines the list of columns to pass as single argument to the indexer operation, which is denoted by outer brackets.

Upvotes: 11

Joseph
Joseph

Reputation: 1136

For pandas objects (Series, DataFrame), the indexing operator [] only accepts

  1. colname or list of colnames to select column(s)
  2. slicing or Boolean array to select row(s), i.e. it only refers to one dimension of the dataframe.

For df[[colname(s)]], the interior brackets are for list, and the outside brackets are indexing operator, i.e. you must use double brackets if you select two or more columns. With one column name, single pair of brackets returns a Series, while double brackets return a dataframe.

Also, df.ix[df['A'] < 3,['A','C']] or df.loc[df['A'] < 3,['A','C']] is better than the chained selection for avoiding returning a copy versus a view of the dataframe.

Please refer pandas documentation for details

Upvotes: 104

Related Questions