登录
首页精彩阅读R语言读取数据、拆分数据,并保存到相应文件夹
R语言读取数据、拆分数据,并保存到相应文件夹
2018-07-06
收藏

R语言读取数据、拆分数据,并保存到相应文件夹

将一张Excel表的数据按照第一列元素分组,分组后按照某种规则命名,并重新读入Excel中,放入指定的文件夹中。

如上所示的excel表,将sheet1中的数据按照“地域名称”拆分,并将数据放入相应的省份文件夹,如放入D盘的test文件夹中
D:\mydata\test
|___mydata.xlsx
|___安徽省
    |___安徽省安庆市.xlsx
    |___安徽省蚌埠市.xlsx
|___河北省
    |__河北省蚌埠市.xlsx
|___……

假如数据最终的文件夹目录如上所示。代码可以作如下编写:

library(readxl)  # 载入readxl包,使用read_excel读入excel文件
library(xlsx)    # 载入xlsx包,使用write.xlsx保存文件为excel格式

# setwd()设置路径为D:\\mydata\\test'
setwd('D:\\mydata\\test')  

# read_excel 读入数据,具体用法可使用帮助。
# 1表示读入第一张sheet,2表示读入第二张sheet,也可用sheet = "sheetname",默认第一行为标题行
dt = read_excel('mydata.xlsx', 2)

# unique() 过滤重复数据,保留唯一数据
# 此处过滤掉第2列和第3列的重复行,最终只剩下3行不重复数据,默认不含标题
pcd <- unique(dt[,2:3])

# pcd 包含7行记录,第7行为空值,将非空值1-6行赋给province_city
province_city <- pcd[1:6,]

setwd('D:\\mydata\\test')   # 设置当前路径为省份文件夹放置的目录
cdir <- setwd('D:\\mydata\\test')

# 循环次数 n=length(dt[,3] 读取所有属于A市的数据
for (i in c(1:length(dt[,3]))){
    # subset()读取子集,subset(data, data[, 1] == "a") 读取data中第一列所有为a的行
    dt1 <- subset(dt, dt[,2]== province_city[i,1])     
    # paste,设置filename, 即安徽省,河北省
    cfilename = paste(province_city[i,1],".xlsx",sep = "")
    # 用于下面的if语句判断 创建的“安徽省”目录是否存在
    f_dir = paste('D:\\mydata\\test\\',province_city[i,2],sep="")

    # 创建的“安徽省”目录不存在,则创建;存在则不动作
    if (file.exists(f_dir) == F){    
     dir.create(province_city[i,2])    
    }else { }

  # 设置文件名称,即安徽省安庆市.xlsx
  c_dir = paste('D:\\mydata\\test\\',province_city[i,2],sep = "")

  # 设置c_dir为当前路径
  setwd(c_dir)

  # write.xlsx 将所需的文件列保存到上述设置的文件中,col.names=TRUE包含标题
  write.xlsx(dt1[,2:5],cfilename,col.names=TRUE,showNA=FALSE)

  # 设置路径为cdir,进入到下一次循环
  setwd(cdir)
}

数据分析咨询请扫描二维码

客服在线
立即咨询