Python - 使用 Pandas.drop() 删除 DataFrame 中的行/列


Pandas 是最流行的用于数据分析和数据整理的 Python 库之一。在本文中,我们将了解如何创建 Pandas DataFrame,然后从中删除一些选定的行或列。

删除行

在下面的示例中,我们有 iris.csv 文件,它被读入 DataFrame。我们首先查看现有的 DataFrame,然后通过提供要删除的值来对索引列应用 drop 函数。正如我们在结果集底部看到的,行数减少了 3。

示例

import pandas as pd
# making data frame from csv file
data = pd.read_csv("E:\iris1.csv",index_col ="Id")
print(data)
# dropping passed values
data.drop([6,9,10],inplace=True)
# display
print(data)

输出

运行以上代码将得到以下结果:

   SepalLengthCm    SepalWidthCm    PetalLengthCm    PetalWidthCm    Species
Id
1    5.1                3.5             1.4             0.2       Iris-setosa
2    4.9                3.0             1.4             0.2       Iris-setosa
3    4.7                3.2             1.3             0.2       Iris-setosa
.   ..   …   .…   .…..……
[150 rows x 5 columns]

After Dropping
   SepalLengthCm    SepalWidthCm    PetalLengthCm    PetalWidthCm    Species
Id
1      5.1                3.5             1.4             0.2       Iris-setosa
2      4.9                3.0             1.4             0.2       Iris-setosa
3      4.7                3.2             1.3             0.2       Iris-setosa
149    6.2                3.4             5.4             2.3       Iris-virginica
150    5.9                3.0             5.1             1.8       Iris-virginica
………………….
[147 rows x 5 columns]

删除列

为了从 Pandas DataFrame 中删除列,我们使用 axis 参数。它的值在 drop 函数中设置为 1,我们提供要删除的列名。正如您所看到的,结果集中的列数从 5 减少到 3。

示例

import pandas as pd
# making data frame from csv file
data = pd.read_csv("E:\iris1.csv",index_col ="Id")
print(data)
# dropping passed values
data.drop(['SepalWidthCm','PetalLengthCm'],axis=1,inplace=True)
print("After Dropping")
# display
print(data)

输出

运行以上代码将得到以下结果:

   SepalLengthCm    SepalWidthCm    PetalLengthCm    PetalWidthCm    Species
Id
1       5.1          3.5                1.4             0.2          Iris-setosa
2       4.9          3.0                1.4             0.2          Iris-setosa
3       4.7          3.2                1.3             0.2          Iris-setosa
.   .   .…   .…   .….   .……
[150 rows x 5 columns]
After Dropping
   SepalLengthCm    PetalWidthCm    Species
Id
1    5.1             0.2             Iris-setosa
2    4.9             0.2             Iris-setosa
3    4.7             0.2             Iris-setosa
.....….
[150 rows x 3 columns]

更新于: 2021年1月25日

790 次浏览

开启您的 职业生涯

通过完成课程获得认证

开始学习
广告