![](/img/trans.png)
[英]How to compare dataframe column to another dataframe column inplace in pyspark?
[英]How to compare values in a pyspark dataframe column with another dataframe in pyspark
我有一個pyspark dataframe(df1)
其第一行如下:
[Row(_c0='{"type":"Fi","values":[0.20100994408130646,1.172734797000885,0.06788740307092667,0.2314232587814331,0.2012220323085785]}', _c1='0')]
我想將“值”列表與下面dataframe(df2)
值的第一列進行比較,如下所示:
0 0.57581 1.25461 0.68694 0.974580 1.54789 0.23646
1 0.98745 0.23655 2.58970 4.587580 0.89756 1.25678
2 0.45780 5.78940 0.65986 2.125400 0.98745 1.23658
3 2.56834 0.25698 4.26587 0.569872 0.36987 0.68975
4 0.25678 1.23654 5.68320 0.986230 0.87563 2.58975
類似地,我在df1
有很多行,我必須看到df1
“values”列表中的哪些值大於df2
的相應列。我需要找到滿足上述條件的那些索引並將其作為列表存儲在另一列中給df1
。
例如1.172737 > 0.98745
所以它的索引是1
1.172737 > 0.98745
我將在df1 named(indices)
有另一列df1 named(indices)
,其中包含value1,如果出現另一個值,它必須附加相同的列。
比較是在各列和行之間。上面顯示的df1行是第1行,因此必須與df2中的第一列進行比較。
如果我沒有得到充分重視......請在評論中告訴我。
此代碼適用於Python 2.7和Spark 2.3.2:
from pyspark.sql import functions as F
from pyspark.sql.types import ArrayType, IntegerType
# Create test dataframes
df1 = spark.createDataFrame([
['{"type":"Fi","values":[0.20100994408130646,1.172734797000885,0.06788740307092667,0.2314232587814331,0.2012220323085785]}', '0'],
['{"type":"Fi","values":[0.6, 0.8, 0.5, 2.1, 0.4]}', '0']
],['_c0','_c1'])
df2 = spark.createDataFrame([
[0, 0.57581, 1.25461, 0.68694, 0.974580, 1.54789, 0.23646],
[1, 0.98745, 0.23655, 2.58970, 4.587580, 0.89756, 1.25678],
[2, 0.45780, 5.78940, 0.65986, 2.125400, 0.98745, 1.23658],
[3, 2.56834, 0.25698, 4.26587, 0.569872, 0.36987, 0.68975],
[4, 0.25678, 1.23654, 5.68320, 0.986230, 0.87563, 2.58975]
],['id','v1', 'v2', 'v3', 'v4', 'v5', 'v6'])
# Get schema and load json correctly
json_schema = spark.read.json(df1.rdd.map(lambda row: row._c0)).schema
df1 = df1.withColumn('json', F.from_json('_c0', json_schema))
# Get column 1 values to compare
values = [row['v1'] for row in df2.select('v1').collect()]
# Define udf to compare values
def cmp_values(lst):
list_cmp = map(lambda t: t[0] > t[1], zip(lst, values)) # Boolean list
return [idx for idx, cond in enumerate(list_cmp) if cond] # Indices of satisfying elements
udf_cmp_values = F.udf(cmp_values, ArrayType(IntegerType()))
# Apply udf on array
df1 = df1.withColumn('indices', udf_cmp_values(df1.json['values']))
df1.show()
+--------------------+---+--------------------+---------+
| _c0|_c1| json| indices|
+--------------------+---+--------------------+---------+
|{"type":"Fi","val...| 0|[Fi, [0.201009944...| [1]|
|{"type":"Fi","val...| 0|[Fi, [0.6, 0.8, 0...|[0, 2, 4]|
+--------------------+---+--------------------+---------+
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.