Question

我的数据框df具有以下结构：

+-----+-----+-----+-------+
|  s  |col_1|col_2|col_...|
+-----+-------------------+
| f1  |  0.0|  0.6|  ...  |
| f2  |  0.6|  0.7|  ...  |
| f3  |  0.5|  0.9|  ...  |
|  ...|  ...|  ...|  ...  |

我想计算这个数据帧的转置，所以它看起来像

+-------+-----+-----+-------+------+
|  s    | f1  | f2  | f3    |   ...|
+-------+-------------------+------+
|col_1  |  0.0|  0.6|  0.5  |   ...|
|col_2  |  0.6|  0.7|  0.9  |   ...|
|col_...|  ...|  ...|  ...  |   ...|

我绑定了这两个解决方案，但它返回的数据帧没有指定的使用方法：

方法1：

 for x in df.columns:
    df = df.pivot(x)

方法2：

df = sc.parallelize([ (k,) + tuple(v[0:]) for k,v in df.items()]).toDF()

我该如何解决这个问题。

Answer 1

如果数据足够小以便转置（不通过聚合转动），您只需将其转换为Pandas DataFrame：

df = sc.parallelize([
    ("f1", 0.0, 0.6, 0.5),
    ("f2", 0.6, 0.7, 0.9)]).toDF(["s", "col_1", "col_2", "col_3"])

df.toPandas().set_index("s").transpose()
s       f1   f2
col_1  0.0  0.6
col_2  0.6  0.7
col_3  0.5  0.9

如果它很大，那么Spark就无济于事。 Spark DataFrame按行分配数据（虽然本地使用柱状存储），因此各行的大小仅限于本地内存。

在Apache Spark中使用pyspark进行数据帧转置

1 个答案: