Pandas删除DataFrame中除某一列以外的所有列
在本文中,我们将介绍如何使用Pandas删除DataFrame中除某一列以外的所有列。这在数据处理中很常见,例如我们可能有一个含有大量列的数据框,但只想关注其中的一列。
阅读更多:Pandas 教程
例子
首先让我们创建一个示例DataFrame:
import pandas as pd
data = {'Name': ['John', 'Peter', 'Sara'],
'Age': [23, 28, 21],
'City': ['New York', 'Paris', 'London']}
df = pd.DataFrame(data)
print(df)
输出:
Name Age City
0 John 23 New York
1 Peter 28 Paris
2 Sara 21 London
现在,我们想删除除City列以外的所有列。我们可以使用drop()函数来实现它。
df.drop(columns=['Name', 'Age'], inplace=True)
print(df)
输出:
City
0 New York
1 Paris
2 London
drop()函数使用columns参数来指定需要删除的列。inplace参数用于在原DataFrame上进行更改还是创建一个新的DataFrame。
另一种方法是使用loc[]函数来选择需要保留的列。
df = pd.DataFrame(data)
df = df.loc[:, ['City']]
print(df)
输出:
City
0 New York
1 Paris
2 London
loc[]函数使用切片来索引一个DataFrame。:表示切片所有行,[...]中的列表指定需要保留的列。
总结
在本文中,我们介绍了如何使用Pandas删除DataFrame中除某一列以外的所有列。我们演示了两种方法:使用drop()函数和loc[]函数。熟练掌握这种操作可以使数据处理更加简便和高效。
极客教程