Planets DataFrame select columns
examples/pandas/planets_select_columns.py
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