将特定选定列作为副本提取到新数据帧

Question

我有一个有4列的pandas DataFrame，我想创建一个只有其中3列的新DataFrame。这个问题类似于：从数据帧中提取特定列，但对pandas来说不是R。下面的代码不起作用，会引发错误，当然也不是pandasnic的方法。

import pandas as pd
old = pd.DataFrame({'A' : [4,5], 'B' : [10,20], 'C' : [100,50], 'D' : [-30,-50]})
new = pd.DataFrame(zip(old.A, old.C, old.D)) # raises TypeError: data argument can't be an iterator 

熊猫式的方法是什么？

Answer 1

另一个更简单的方法似乎是：

new = pd.DataFrame([old.A, old.B, old.C]).transpose()

其中old.column_name将给出一个系列。列出要保留的所有列序列，并将其传递给DataFrame构造函数。我们需要做一个换位来调整形状。

In [14]:pd.DataFrame([old.A, old.B, old.C]).transpose()
Out[14]: 
   A   B    C
0  4  10  100
1  5  20   50

Answer 2

最简单的方法是

new = old[['A','C','D']]

.

Answer 3

有一种方法可以这样做，实际上它看起来类似于R

new = old[['A', 'C', 'D']].copy()

在这里，您只是从原始数据帧中选择所需的列，并为这些列创建一个变量。如果您想要修改新的dataframe，您可能希望使用.copy()来避免SettingWithCopyWarning。

另一种方法是使用filter，它将在默认情况下创建副本：

new = old.filter(['A','B','D'], axis=1)

最后，根据原始数据帧中的列数，使用drop（默认情况下，这也将创建一个副本）来表达它可能更简洁：

new = old.drop('B', axis=1)