pandas get column of dataframe code example
Example 1: python - subset specific columns name in a dataframe
columns = ['b', 'c']
df1 = pd.DataFrame(df, columns=columns)
Example 2: select columns pandas
df1 = df.iloc[:,0:2] # Remember that Python does not slice inclusive of the ending index.
Example 3: add an index column in range dataframe
df = df.loc[df.index.repeat(df['a'])]
df['c'] = df.groupby(level=0).cumcount() + 1
df = df.reset_index(drop=True)
print (df)
a b c
0 1 x 1
1 2 y 1
2 2 y 2
3 3 z 1
4 3 z 2
5 3 z 3
Example 4: select subset of columns from dataframe
In [25]: titanic.iloc[9:25, 2:5]
Out[25]:
Pclass Name Sex
9 2 Nasser, Mrs. Nicholas (Adele Achem) female
10 3 Sandstrom, Miss. Marguerite Rut female
11 1 Bonnell, Miss. Elizabeth female
12 3 Saundercock, Mr. William Henry male
13 3 Andersson, Mr. Anders Johan male
.. ... ... ...
20 2 Fynney, Mr. Joseph J male
21 2 Beesley, Mr. Lawrence male
22 3 McGowan, Miss. Anna "Annie" female
23 1 Sloper, Mr. William Thompson male
24 3 Palsson, Miss. Torborg Danira female
[16 rows x 3 columns]