select columns in pandas by index code example
Example 1: pandas select column by index
# A B C
# 0 1 3 5
# 1 2 4 6
column_B = a_dataframe.iloc[:, 1]
print(column_B)
# OUTPUT
# 0 3
# 1 4
Example 2: select columns in pandas df
In [8]: age_sex = titanic[["Age", "Sex"]]
In [9]: age_sex.head()
Out[9]:
Age Sex
0 22.0 male
1 38.0 female
2 26.0 female
3 35.0 female
4 35.0 male