Python Pandas：将选定的列保留为DataFrame而不是Series

Python Pandas：将选定的列保留为DataFrame而不是Series

正如@Jeff提到的，有几种方法可以做到这一点，但我建议使用loc / iloc来使其更明确（如果尝试歧义，请提早出错）：

In [10]: df = pd.Dataframe([[1, 2], [3, 4]], columns=['A', 'B'])In [11]: dfOut[11]:   A  B0  1  21  3  4In [12]: df[['A']]In [13]: df[[0]]In [14]: df.loc[:, ['A']]In [15]: df.iloc[:, [0]]Out[12-15]:  # they all return the same thing:   A0  11  3

在整数列名称的情况下，后两种选择消除了歧义（正是创建loc / iloc的原因）。例如：

In [16]: df = pd.Dataframe([[1, 2], [3, 4]], columns=['A', 0])In [17]: dfOut[17]:   A  00  1  21  3  4In [18]: df[[0]]  # ambiguousOut[18]:   A0  11  3