import sys
import pandas as pd
filename = "planets.csv"
if len(sys.argv) == 2:
filename = sys.argv[1]
df = pd.read_csv(filename)
mass = df['Mass'] # Get Series
print(type(mass))
print(mass)
print()
one_col = df[['Mass']] # Get DataFrame
print(type(one_col))
print(one_col)
print()
two_cols = df[['Planet name', 'Mass']] # Select multiple columns, get DataFrame
print(type(two_cols))
print(two_cols)
print()
<class 'pandas.core.series.Series'>
0 0.055000
1 0.815000
2 1.000000
3 0.107000
4 0.000150
5 318.000000
6 95.000000
7 14.000000
8 17.000000
9 0.002180
10 0.000254
Name: Mass, dtype: float64
<class 'pandas.core.frame.DataFrame'>
Mass
0 0.055000
1 0.815000
2 1.000000
3 0.107000
4 0.000150
5 318.000000
6 95.000000
7 14.000000
8 17.000000
9 0.002180
10 0.000254
<class 'pandas.core.frame.DataFrame'>
Planet name Mass
0 Mercury 0.055000
1 Venus 0.815000
2 Earth 1.000000
3 Mars 0.107000
4 Ceres 0.000150
5 Jupiter 318.000000
6 Saturn 95.000000
7 Uranus 14.000000
8 Neptune 17.000000
9 Pluto 0.002180
10 Charon 0.000254