Python Pandas DataFrame squeeze方法用法及代码示例

例子

挤压single-column DataFrame

考虑以下 DataFrame ：

df = pd.DataFrame({"A":[3,4]})
df



   A
0  3
1  4

由于我们的 DataFrame 仅包含单个列，因此我们可以将其简化为 Series，如下所示：

df.squeeze()



A    3
B    4
Name: 0, dtype: int64

挤压单行DataFrame

考虑以下 DataFrame ：

df = pd.DataFrame({"A":[3],"B":[4]})
df



   A  B
0  3  4

由于我们的 DataFrame 仅包含一行，因此我们可以将其简化为 Series，如下所示：

df.squeeze()



A    3
B    4
Name: 0, dtype: int64

指定轴参数

默认情况下，会检查行和列以查看是否可以进行减少。我们可以通过指定 axis 参数将此检查限制为仅行或列。

例如，考虑以下 DataFrame：

df = pd.DataFrame({"A":[3],"B":[4]})
df



   A  B
0  3  4

如果我们尝试使用列进行挤压，我们只会得到源DataFrame df：

df.squeeze("columns")



   A  B
0  3  4