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