R - CSV 文件

在 R 中，我们可以从存储在 R 环境之外的文件读取数据。我们还可以将数据写入文件，由操作系统存储和访问。R 可以读取和写入各种文件格式，如 csv、excel、xml 等。

在本章中，我们将学习如何从 csv 文件中读取数据，然后将数据写入 csv 文件。该文件应存在于当前工作目录中，以便 R 可以读取它。当然，我们也可以设置自己的目录并从那里读取文件。

获取和设置工作目录

您可以使用 getwd（） 函数检查 R 工作区指向哪个目录。您还可以使用 setwd（） 函数设置新的工作目录。

# 获取并打印当前工作目录。
print(getwd())

# 设置当前工作目录。
setwd("/web/com")

# 获取并打印当前工作目录。
print(getwd())

当我们执行上述代码时，它会产生以下结果——

此结果取决于您的操作系统和您正在工作的当前目录。

输入为 CSV 文件

csv 文件是一个文本文件，其中列中的值用逗号分隔。让我们考虑名为 input.csv 的文件中存在的以下数据。

您可以通过复制和粘贴此数据来使用 Windows 记事本创建此文件。使用记事本中的 Save As All files（*.*） 选项将文件另存为 input.csv。

读取 CSV 文件

以下是 read.csv（） 函数的简单示例，用于读取当前工作目录中可用的 CSV 文件 -

data <- read.csv("input.csv")
print(data)

当我们执行上述代码时，它会产生以下结果——

分析 CSV 文件

默认情况下，read.csv（） 函数将输出作为数据帧提供。这可以很容易地检查如下。此外，我们还可以检查列数和行数。

data <- read.csv("input.csv")

print(is.data.frame(data))
print(ncol(data))
print(nrow(data))

当我们执行上述代码时，它会产生以下结果——

一旦我们在数据帧中读取数据，我们就可以应用适用于数据帧的所有函数，如后续部分所述。

获得最高薪水

# 创建数据帧。
data <- read.csv("input.csv")

# 从数据框中获取最高工资。
sal <- max(data$salary)
print(sal)

当我们执行上述代码时，它会产生以下结果——

获取最高薪水人士的详细信息

我们可以获取满足特定过滤条件的行，类似于 SQL where 子句。

# 创建数据帧。
data <- read.csv("input.csv")

# 从数据框中获取最高工资。
sal <- max(data$salary)

# 获取有最高工资的人员详细信息。
retval <- subset(data, salary == max(salary))
print(retval)

当我们执行上述代码时，它会产生以下结果——

让 IT 部门的所有员工都参与进来

# 创建数据帧。
data <- read.csv("input.csv")

retval <- subset( data, dept == "IT")
print(retval)

当我们执行上述代码时，它会产生以下结果——

IT 获取薪资大于 600 的 IT 部门人员

# 创建数据帧。
data <- read.csv("input.csv")

info <- subset(data, salary > 600 & dept == "IT")
print(info)

当我们执行上述代码时，它会产生以下结果——

IT 获取 2014 年当天或之后加入的人员

# 创建数据帧。
data <- read.csv("input.csv")

retval <- subset(data, as.Date(start_date) > as.Date("2014-01-01"))
print(retval)

当我们执行上述代码时，它会产生以下结果——

写入 CSV 文件

R 可以从现有数据框创建 csv 文件。write.csv（） 函数用于创建 csv 文件。此文件在工作目录中创建。

# 创建数据帧。
data <- read.csv("input.csv")
retval <- subset(data, as.Date(start_date) > as.Date("2014-01-01"))

# 将筛选后的数据写入新文件。
write.csv(retval,"output.csv")
newdata <- read.csv("output.csv")
print(newdata)

当我们执行上述代码时，它会产生以下结果——

此处的 X 列来自数据集 newper。可以在写入文件时使用其他参数删除此参数。

# 创建数据帧。
data <- read.csv("input.csv")
retval <- subset(data, as.Date(start_date) > as.Date("2014-01-01"))

# 将筛选后的数据写入新文件。
write.csv(retval,"output.csv", row.names = FALSE)
newdata <- read.csv("output.csv")
print(newdata)

当我们执行上述代码时，它会产生以下结果——