根据字典选择所有行和列

Question

所以我有一个大型数据框，其中包含一系列基于状态的信息。我只想使用某些状态，我把它放入字典中。

我把索引放在那里，因为状态的名称在该列中。有没有办法只使用状态名称选择所有行和列，以及它具有的信息。我认为我使用的方式不会选择所有信息。\

Answer 1

我已下载a freely available CSV file with state data并将其保存到states.csv。我可以选择给出如下状态列表的行：

import pandas as pd

states = ['New York', 'California']

df = pd.read_csv('states.csv')
df['Population'] = df['Population est.'].str.split(',').str.join('').astype(int)

df1 = df.loc[df['State'].isin(states)]

print df1[['State', 'Capital', 'Population']]

<强>输出

         State     Capital  Population
4   California  Sacramento    38041430
31    New York      Albany    19570261