java通过匹配合并数据(数据预处理)

数据描述

以下程序是对如下格式的数据进行合并处理。

这里写图片描述

这个表的每一行表示用户id及用户的特征。其中,一个用户只有一个特征向量,即第一列不会重复。

这里写图片描述
这张表的第一列,表示用户的id,第二列表示用户所看的电影,第三列表示用户对电影的打分(1-13分),第四列表示用户对电影的打分,但分值范围是1-5分。

问题描述

在做数据预处理时,如何将第二张表添加上用户特征呢?其实,方法很简单,将第二张表的用户id与第一张表的用户id进行匹配就行。合并结果如下图所示。
这里写图片描述

数据处理程序

package deal;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/*
 * author:合肥工业大学 管院学院 钱洋 
 * email:1563178220@qq.com
*/

public class GetPUser {
    public static List<String> readDocs(String docsPath,String code) throws IOException{
         BufferedReader reader = new BufferedReader( new InputStreamReader( new FileInputStream( new File(docsPath)),code));
         String s=null;
         List<String> userproductscore=new ArrayList<String>();
         while ((s=reader.readLine())!=null) {
             userproductscore.add(s);
         }
         reader.close();
         return userproductscore;
    }
    public static HashMap<String, String> MAPread(String docsPath1,String code1) throws IOException{
        BufferedReader reader1 = new BufferedReader( new InputStreamReader( new FileInputStream( new File(docsPath1)),code1));
         String s1=null;
         HashMap<String,String> userfeaturemap=new HashMap<String,String>();
         while ((s1=reader1.readLine())!=null) {
             String arr[]=s1.split("\t");
             String feature="";
             for (int i = 1; i < arr.length; i++) {
                 BigDecimal db = new BigDecimal(arr[i]);
                 String ii = db.toPlainString();
                 feature+=ii+" ";
            }

             userfeaturemap.put(s1.split("\t")[0], feature);
         }
         reader1.close();
         return userfeaturemap;
    }

    public static List<String> match(List<String> userproductscore,HashMap<String, String> userfeaturemap) throws IOException{
        List<String> userscoreandfeature=new ArrayList<>();
        for (int i = 0; i < userproductscore.size(); i++) {
            //获取用户id
            String user_id=userproductscore.get(i).split("\t")[0];
            //获取用户特征
            String userfeature = userfeaturemap.get(user_id);
            userscoreandfeature.add(userproductscore.get(i)+"\t"+userfeature);
            System.out.println(userproductscore.get(i)+"\t"+userfeature);
        }
        return userscoreandfeature;
    }
    public static void main(String[] args) throws IOException {  
        //读取两个文本
        List<String> userproductscore=readDocs("data/train/ydata-ymovies-user-movie-ratings-train-v1_0.txt","gbk");
        HashMap<String, String> userfeaturemap=MAPread("data/fileofuser/yahoo.txt","utf-8");
        //匹配结果
        match(userproductscore,userfeaturemap);
    }  
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值