import pandas
food_info = pandas.read_csv("food_info.csv")
print(type(food_info))
print(food_info.columns)
print(food_info.head(2))
col_names = food_info.columns.tolist()
gram_columns = []
for c in col_names:
if c.endswith("(g)"):
gram_columns.append(c)
gram_df = food_info[gram_columns]
print(gram_df.head(3))