import sys
import pandas as pd
filename = "planets.csv"
if len(sys.argv) == 2:
filename = sys.argv[1]
df = pd.read_csv(filename)
rows = df[2:5] # using row numbers
print(type(rows))
print(rows)
print()
iloc_rows = df.iloc[2:5] # just like plain []
print(type(iloc_rows))
print(iloc_rows)
print()
picked_rows = df.iloc[[2,5,3]] # using specific row numbers
print(type(picked_rows))
print(picked_rows)
print()
# df[[2,5,3]] would not work so we need iloc
<class 'pandas.core.frame.DataFrame'>
Planet name Distance (AU) Mass
2 Earth 1.00 1.00000
3 Mars 1.50 0.10700
4 Ceres 2.77 0.00015
<class 'pandas.core.frame.DataFrame'>
Planet name Distance (AU) Mass
2 Earth 1.00 1.00000
3 Mars 1.50 0.10700
4 Ceres 2.77 0.00015
<class 'pandas.core.frame.DataFrame'>
Planet name Distance (AU) Mass
2 Earth 1.0 1.000
5 Jupiter 5.2 318.000
3 Mars 1.5 0.107