连接两个数据框,从一个框中选择所有列,从另一个框中选择一些列

ttcibm8c  于 2022-09-21  发布在  Spark
关注(0)|答案(12)|浏览(205)

假设我具有具有几列(其中列id)的Spark Dataframe df1和具有两列idother的 Dataframe df2

有没有办法复制以下命令:

sqlContext.sql("SELECT df1.*, df2.other FROM df1 JOIN df2 ON df1.id = df2.id")

通过仅使用诸如join()select()等Spark源函数?

我必须在函数中实现此联接,并且我不希望被强制将sqlContext作为函数参数。

nszi6y05

nszi6y051#

星号(*)适用于别名。例:

from pyspark.sql.functions import *

df1 = df1.alias('df1')
df2 = df2.alias('df2')

df1.join(df2, df1.id == df2.id).select('df1.*')
wlsrxk51

wlsrxk512#

我不确定这是不是最有效的方法,但这对我很管用:

from pyspark.sql.functions import col

df1.alias('a').join(df2.alias('b'),col('b.id') == col('a.id')).select([col('a.'+xx) for xx in a.columns] + [col('b.other1'),col('b.other2')])

诀窍在于:

[col('a.'+xx) for xx in a.columns] : all columns in a

[col('b.other1'),col('b.other2')] : some columns of b
yvgpqqbh

yvgpqqbh3#

不使用别名。

df1.join(df2, df1.id == df2.id).select(df1["*"],df2["other"])
nwo49xxi

nwo49xxi4#

这是一个不需要SQL上下文,但维护DataFrame的元数据的解决方案。

a = sc.parallelize([['a', 'foo'], ['b', 'hem'], ['c', 'haw']]).toDF(['a_id', 'extra'])
b = sc.parallelize([['p1', 'a'], ['p2', 'b'], ['p3', 'c']]).toDF(["other", "b_id"])

c = a.join(b, a.a_id == b.b_id)

然后,c.show()产生:

+----+-----+-----+----+
|a_id|extra|other|b_id|
+----+-----+-----+----+
|   a|  foo|   p1|   a|
|   b|  hem|   p2|   b|
|   c|  haw|   p3|   c|
+----+-----+-----+----+
kb5ga3dv

kb5ga3dv5#

我相信这将是最简单和最直观的方式:

final = (df1.alias('df1').join(df2.alias('df2'),
                               on = df1['id'] == df2['id'],
                               how = 'inner')
                         .select('df1.*',
                                 'df2.other')
)
8wtpewkr

8wtpewkr6#

删除重复的b_id

c = a.join(b, a.a_id == b.b_id).drop(b.b_id)
shyt4zoc

shyt4zoc7#

下面是代码片段,它执行内部联接,并从dataframe中选择列,并将同一列的别名设置为不同的列名。

emp_df  = spark.read.csv('Employees.csv', header =True);
dept_df = spark.read.csv('dept.csv', header =True)

emp_dept_df = emp_df.join(dept_df,'DeptID').select(emp_df['*'], dept_df['Name'].alias('DName'))
emp_df.show()
dept_df.show()
emp_dept_df.show()
Output  for 'emp_df.show()':

+---+---------+------+------+
| ID|     Name|Salary|DeptID|
+---+---------+------+------+
|  1|     John| 20000|     1|
|  2|    Rohit| 15000|     2|
|  3|    Parth| 14600|     3|
|  4|  Rishabh| 20500|     1|
|  5|    Daisy| 34000|     2|
|  6|    Annie| 23000|     1|
|  7| Sushmita| 50000|     3|
|  8| Kaivalya| 20000|     1|
|  9|    Varun| 70000|     3|
| 10|Shambhavi| 21500|     2|
| 11|  Johnson| 25500|     3|
| 12|     Riya| 17000|     2|
| 13|    Krish| 17000|     1|
| 14| Akanksha| 20000|     2|
| 15|   Rutuja| 21000|     3|
+---+---------+------+------+

Output  for 'dept_df.show()':
+------+----------+
|DeptID|      Name|
+------+----------+
|     1|     Sales|
|     2|Accounting|
|     3| Marketing|
+------+----------+

Join Output:
+---+---------+------+------+----------+
| ID|     Name|Salary|DeptID|     DName|
+---+---------+------+------+----------+
|  1|     John| 20000|     1|     Sales|
|  2|    Rohit| 15000|     2|Accounting|
|  3|    Parth| 14600|     3| Marketing|
|  4|  Rishabh| 20500|     1|     Sales|
|  5|    Daisy| 34000|     2|Accounting|
|  6|    Annie| 23000|     1|     Sales|
|  7| Sushmita| 50000|     3| Marketing|
|  8| Kaivalya| 20000|     1|     Sales|
|  9|    Varun| 70000|     3| Marketing|
| 10|Shambhavi| 21500|     2|Accounting|
| 11|  Johnson| 25500|     3| Marketing|
| 12|     Riya| 17000|     2|Accounting|
| 13|    Krish| 17000|     1|     Sales|
| 14| Akanksha| 20000|     2|Accounting|
| 15|   Rutuja| 21000|     3| Marketing|
+---+---------+------+------+----------+
uqjltbpv

uqjltbpv8#

我收到一个错误:‘A Not Found’,使用建议的代码:

from pyspark.sql.functions import col df1.alias('a').join(df2.alias('b'),col('b.id') == col('a.id')).select([col('a.'+xx) for xx in a.columns] + [col('b.other1'),col('b.other2')])

我将a.columns更改为df1.columns,结果成功了。

w7t8yxp5

w7t8yxp59#

加入后丢弃重复列的功能。

查看

Def dropDupeDfCols(Df):新数据集=[]数据集=[]

for i in range(len(df.columns)):
    if df.columns[i] not in newcols:
        newcols.append(df.columns[i])
    else:
        dupcols.append(i)

df = df.toDF(*[str(i) for i in range(len(df.columns))])
for dupcol in dupcols:
    df = df.drop(str(dupcol))

return df.toDF(*newcols)
mbjcgjjk

mbjcgjjk10#

我只是删除了df2中不需要的专栏,并加入了:

sliced_df = df2.select(columns_of_interest)
df1.join(sliced_df, on=['id'], how='left')

**id should be in `columns_of_interest` tho
avwztpqn

avwztpqn11#

df1.join(df2, ['id']).drop(df2.id)
brccelvz

brccelvz12#

您可以只进行连接,然后选择所需的列https://spark.apache.org/docs/latest/api/python/pyspark.sql.html?highlight=dataframe%20join#pyspark.sql.DataFrame.join

相关问题