當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文

Python pyspark DataFrame.unstack用法及代碼示例

本文簡要介紹 pyspark.pandas.DataFrame.unstack 的用法。

用法:

DataFrame.unstack() → Union[DataFrame, Series]

旋轉(必要的分層)索引標簽。

返回具有新級別列標簽的DataFrame，其最內層由旋轉索引標簽組成。

如果索引不是 MultiIndex，則輸出將是一個係列。

注意

如果索引是 MultiIndex，則輸出 DataFrame 可能非常寬，並且可能會導致嚴重的性能下降，因為 Spark 基於行進行分區。

返回：

係列或DataFrame

例子：

>>> df = ps.DataFrame({"A": {"0": "a", "1": "b", "2": "c"},
...                    "B": {"0": "1", "1": "3", "2": "5"},
...                    "C": {"0": "2", "1": "4", "2": "6"}},
...                   columns=["A", "B", "C"])
>>> df
   A  B  C
0  a  1  2
1  b  3  4
2  c  5  6
>>> df.unstack().sort_index()
A  0    a
   1    b
   2    c
B  0    1
   1    3
   2    5
C  0    2
   1    4
   2    6
dtype: object
>>> df.columns = pd.MultiIndex.from_tuples([('X', 'A'), ('X', 'B'), ('Y', 'C')])
>>> df.unstack().sort_index()
X  A  0    a
      1    b
      2    c
   B  0    1
      1    3
      2    5
Y  C  0    2
      1    4
      2    6
dtype: object

對於MultiIndex案例：

>>> df = ps.DataFrame({"A": ["a", "b", "c"],
...                    "B": [1, 3, 5],
...                    "C": [2, 4, 6]},
...                   columns=["A", "B", "C"])
>>> df = df.set_index('A', append=True)
>>> df  
     B  C
  A
0 a  1  2
1 b  3  4
2 c  5  6
>>> df.unstack().sort_index()  
     B              C
A    a    b    c    a    b    c
0  1.0  NaN  NaN  2.0  NaN  NaN
1  NaN  3.0  NaN  NaN  4.0  NaN
2  NaN  NaN  5.0  NaN  NaN  6.0

相關用法


注：本文由純淨天空篩選整理自spark.apache.org大神的英文原創作品 pyspark.pandas.DataFrame.unstack。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。