pyspark:如何在数据帧中重复n次? [英] Pyspark: how to duplicate a row n time in dataframe?
本文介绍了pyspark:如何在数据帧中重复n次?的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!
问题描述
我有一个这样的数据框,如果列n
大于一,我想将行重复n次:
I've got a dataframe like this and I want to duplicate the row n times if the column n
is bigger than one:
A B n
1 2 1
2 9 1
3 8 2
4 1 1
5 3 3
并像这样转换:
A B n
1 2 1
2 9 1
3 8 2
3 8 2
4 1 1
5 3 3
5 3 3
5 3 3
我认为我应该使用explode
,但是我不明白它的工作原理...
谢谢
I think I should use explode
, but I don't understand how it works...
Thanks
推荐答案
一种利用此功能的方法是使用udf
为每一行创建大小为n
的列表.然后爆炸生成的数组.
One way to exploit this function is to use a udf
to create a list of size n
for each row. Then explode the resulting array.
from pyspark.sql.functions import udf, explode
from pyspark.sql.types import ArrayType, IntegerType
df = spark.createDataFrame([(1,2,1), (2,9,1), (3,8,2), (4,1,1), (5,3,3)] ,["A", "B", "n"])
# use udf function to transform the n value to n times
n_to_array = udf(lambda n : [n] * n, ArrayType(IntegerType()))
df2 = df.withColumn('n', n_to_array(df.n))
# now use explode
df2.withColumn('n', explode(df2.n)).show()
+---+---+---+
| A | B | n |
+---+---+---+
| 1| 2| 1|
| 2| 9| 1|
| 3| 8| 2|
| 3| 8| 2|
| 4| 1| 1|
| 5| 3| 3|
| 5| 3| 3|
| 5| 3| 3|
+---+---+---+
这篇关于pyspark:如何在数据帧中重复n次?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!
查看全文