通过反射RDD2DataFrame

java版本:
import org.apache.spark.SparkConf;
import org.apache.spark.api.java.JavaRDD;
import org.apache.spark.api.java.JavaSparkContext;
import org.apache.spark.api.java.function.Function;
import org.apache.spark.sql.DataFrame;
import org.apache.spark.sql.Row;
import org.apache.spark.sql.SQLContext;

import java.util.List;

/**
* Created by rong on 2016/3/19.
*/
public class RDD2DataFrame {
public static void main(String[] args) {
SparkConf conf = new SparkConf().setAppName(“RDD2DataFrame”).setMaster(“local”);
JavaSparkContext sc= new JavaSparkContext(conf);

    JavaRDD<String> testFile = sc.textFile("C://Users//rong//Desktop//persons.txt");

    SQLContext sqlContext = new SQLContext(sc);

    JavaRDD<Person> persons = testFile.map(new Function<String, Person>() {
        public Person call(String line) throws Exception {
            String[]  str = line.split(",");
            Person p = new Person();
            p.setId(Integer.valueOf(str[0]));
            p.setName(str[1]);
            p.setAge(Integer.valueOf(str[2]));
            return p;
        }
    });

    //通过反射技术根据Person.class文件生成DataFrame
    DataFrame df = sqlContext.createDataFrame(persons, Person.class);
    df.registerTempTable("persons");
    df.show();//相当于select * from persons;
    df.select(df.col("name")).show();//相当于select name from person
    df.select(df.col("id"),df.col("name")).show();//相当于select id,name from person
    df.filter(df.col("age").gt(6)).show();

    DataFrame dfs = sqlContext.sql("select * from persons where age >= 6");
    JavaRDD<Row> row = dfs.javaRDD();
    JavaRDD<Person> personRdd = row.map(new Function<Row, Person>() {
        public Person call(Row row) throws Exception {
            Person p = new Person();
            p.setId(row.getInt(1));
            p.setName(row.getString(2));
            p.setAge(row.getInt(0));
            return p;
        }
    });
    List<Person> list = personRdd.collect();
    for(Person p : list){
        System.out.println(p);
    }
}

}

Person类:
person类需要实现Serializable序列化接口,并且是public的。
import java.io.Serializable;

/**
* Created by rong on 2016/3/19.
*/
public class Person implements Serializable {
private int id;
private String name;
private int age;

public int getId() {
    return id;
}

public void setId(int id) {
    this.id = id;
}

public String getName() {
    return name;
}

public void setName(String name) {
    this.name = name;
}

public int getAge() {
    return age;
}

public void setAge(int age) {
    this.age = age;
}

@Override
public String toString() {
    return "Person{" +
            "id=" + id +
            ", name='" + name + '\'' +
            ", age=" + age +
            '}';
}

}

scala版本:
import org.apache.spark.sql.SQLContext
import org.apache.spark.{SparkConf, SparkContext}

/**
* Created by rong on 2016/3/19.
*/
object RDD2DataFrameByScala {

def main(args: Array[String]) {
val conf = new SparkConf().setAppName(“RDD2DataFrameByScala”).setMaster(“local”)
val sc = new SparkContext(conf)
val sqlContext = new SQLContext(sc)
import sqlContext.implicits._
val df = sc.textFile(“C://Users//rong//Desktop//persons.txt”).map(line => line.split(“,”)).map(line => Person(Integer.valueOf(line(0)),String.valueOf(line(1)),Integer.valueOf(line(2)))).toDF
df.show()
}
}

case class Person(id:Int,name:String,age:Int)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值