欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

pyspark.sql.DataFrame与pandas.DataFrame之间的相互转换实例

程序员文章站 2024-01-20 14:36:46
代码如下,步骤流程在代码注释中可见: # -*- coding: utf-8 -*- import pandas as pd from pyspark.sql...

代码如下,步骤流程在代码注释中可见:

# -*- coding: utf-8 -*-
import pandas as pd
from pyspark.sql import sparksession
from pyspark.sql import sqlcontext
from pyspark import sparkcontext
 
#初始化数据
 
#初始化pandas dataframe
df = pd.dataframe([[1, 2, 3], [4, 5, 6]], index=['row1', 'row2'], columns=['c1', 'c2', 'c3'])
 
#打印数据
print df
 
#初始化spark dataframe
sc = sparkcontext()
if __name__ == "__main__":
 spark = sparksession\
  .builder\
  .appname("testdataframe")\
  .getorcreate()
 
sentencedata = spark.createdataframe([
 (0.0, "i like spark"),
 (1.0, "pandas is useful"),
 (2.0, "they are coded by python ")
], ["label", "sentence"])
 
#显示数据
sentencedata.select("label").show()
 
#spark.dataframe 转换成 pandas.dataframe
sqlcontest = sqlcontext(sc)
spark_df = sqlcontest.createdataframe(df)
 
#显示数据
spark_df.select("c1").show()
 
 
# pandas.dataframe 转换成 spark.dataframe
pandas_df = sentencedata.topandas()
 
#打印数据
print pandas_df

程序结果:

pyspark.sql.DataFrame与pandas.DataFrame之间的相互转换实例

以上这篇pyspark.sql.dataframe与pandas.dataframe之间的相互转换实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。