df.iloc[:, lambda x : [0,3]] code example

Example: iloc python

>>> df.iloc[[0, 1]]
     a    b    c    d
0    1    2    3    4
1  100  200  300  400