sqldata = hiveSql.sql("""
        SELECT name,age from DB
""")

tabel_df = pd.DataFrame(data=sqldata.collect())
tabel_df.to_csv('resultRDD.csv')

Как показывает код, у меня есть CSV, как показано ниже:

tom  16
jack 18

На самом деле, мне нужен CSV, как показано ниже

name age
tom  16
jack 18
1
yang xu 21 Дек 2019 в 06:31

4 ответа

Лучший ответ

Просто создайте список имен столбцов:

table_df = pd.Dataframe(data=sqldata.collect())

#add these 2 lines
column_names = ['name','age']
table_df.columns = column_names

table_df.to_csv('resultRDD.csv')

Вот и все. Это сделано!

1
Strange 21 Дек 2019 в 04:10

Вы также можете передать список имен столбцов, таких как

tabel_df = pd.DataFrame(data=sqldata.collect(),columns=['name','age'])
0
Marcin Orlowski 21 Дек 2019 в 04:19

Использовать это:

tabel_df.to_csv('resultRDD.csv', header=['name','age'])
0
oppressionslayer 21 Дек 2019 в 04:27

По-своему

tabel_df = pd.DataFrame(data=sqldata.collect(),columns=sqldata.columns)

Также может работать, Thx ALL

0
yang xu 21 Дек 2019 в 06:29