r4ds/data-import.Rmd

312 lines
14 KiB
Plaintext
Raw Normal View History

2021-02-22 19:47:39 +08:00
# Data import {#data-import}
2015-12-12 02:34:20 +08:00
2021-05-04 21:10:39 +08:00
```{r, results = "asis", echo = FALSE}
status("restructuring")
```
<!--# TO DO: This chapter got moved here from the wrangle section, make sure it makes sense in this new location, doesn't assume anything that comes after it. -->
2016-07-07 02:59:50 +08:00
## Introduction
2016-01-27 21:33:35 +08:00
Working with data provided by R packages is a great way to learn the tools of data science, but at some point you want to stop learning and start working with your own data.
In this chapter, you'll learn how to read plain-text rectangular files into R.
Here, we'll only scratch the surface of data import, but many of the principles will translate to other forms of data.
We'll finish with a few pointers to packages that are useful for other types of data.
2016-01-27 21:33:35 +08:00
2016-07-07 02:59:50 +08:00
### Prerequisites
2015-09-23 02:35:39 +08:00
In this chapter, you'll learn how to load flat files in R with the **readr** package, which is part of the core tidyverse.
2015-09-23 02:35:39 +08:00
2016-10-04 01:30:24 +08:00
```{r setup, message = FALSE}
library(tidyverse)
2016-07-07 02:59:50 +08:00
```
2015-09-23 02:35:39 +08:00
2016-07-12 04:38:39 +08:00
## Getting started
2015-09-23 02:35:39 +08:00
Most of readr's functions are concerned with turning flat files into data frames:
- `read_csv()` reads comma delimited files, `read_csv2()` reads semicolon separated files (common in countries where `,` is used as the decimal place), `read_tsv()` reads tab delimited files, and `read_delim()` reads in files with any delimiter.
2015-09-23 02:35:39 +08:00
- `read_fwf()` reads fixed width files.
You can specify fields either by their widths with `fwf_widths()` or their position with `fwf_positions()`.
`read_table()` reads a common variation of fixed width files where columns are separated by white space.
2015-09-23 02:35:39 +08:00
- `read_log()` reads Apache style log files.
(But also check out [webreadr](https://github.com/Ironholds/webreadr) which is built on top of `read_log()` and provides many more helpful tools.)
2015-09-23 02:35:39 +08:00
These functions all have similar syntax: once you've mastered one, you can use the others with ease.
For the rest of this chapter we'll focus on `read_csv()`.
Not only are csv files one of the most common forms of data storage, but once you understand `read_csv()`, you can easily apply your knowledge to all the other functions in readr.
2015-09-23 02:35:39 +08:00
## Reading data from a file
Here is what a simple CSV file with a row for column names (also commonly referred to as the header row) and six rows of data looks like.
```{r echo = FALSE, message = FALSE}
2022-02-24 03:15:52 +08:00
read_lines("data/students.csv") |> cat(sep = "\n")
```
Note that the `,`s separate the columns.
Table \@ref(tab:students-table) shows a representation of the same data as a table.
```{r students-table, echo = FALSE, message = FALSE}
2022-02-24 03:15:52 +08:00
read_csv("data/students.csv") |>
knitr::kable(caption = "Data from the students.csv file as a table.")
```
2016-08-12 21:09:18 +08:00
The first argument to `read_csv()` is the most important: it's the path to the file to read.
2015-09-23 02:35:39 +08:00
2016-07-27 03:57:25 +08:00
```{r, message = TRUE}
students <- read_csv("data/students.csv")
2016-07-07 02:59:50 +08:00
```
2015-09-23 02:35:39 +08:00
When you run `read_csv()` it prints out a message that tells you how many rows (excluding the header row) and columns the data has along with the delimiter used, and the column specifications (names of columns organized by the type of data the column contains).
It also prints out some information about how to retrieve the full column specification as well as how to quiet this message.
This message is an important part of readr, which we'll come back to in Section \@ref(parsing-a-file) on parsing a file.
2015-09-23 02:35:39 +08:00
You can also supply an inline csv file.
This is useful for experimenting with readr and for creating reproducible examples to share with others:
2015-09-23 02:35:39 +08:00
```{r message = FALSE}
2016-07-07 02:59:50 +08:00
read_csv("a,b,c
1,2,3
4,5,6")
```
2016-01-27 21:33:35 +08:00
In both cases `read_csv()` uses the first line of the data for the column names, which is a very common convention.
2022-03-05 12:58:23 +08:00
There are two cases where you might want to tweak this behavior:
1. Sometimes there are a few lines of metadata at the top of the file.
You can use `skip = n` to skip the first `n` lines; or use `comment = "#"` to drop all lines that start with (e.g.) `#`.
2015-09-23 02:35:39 +08:00
```{r message = FALSE}
2016-07-08 00:17:11 +08:00
read_csv("The first line of metadata
The second line of metadata
2016-07-07 02:59:50 +08:00
x,y,z
2016-07-08 00:17:11 +08:00
1,2,3", skip = 2)
2016-07-07 02:59:50 +08:00
read_csv("# A comment I want to skip
x,y,z
1,2,3", comment = "#")
```
2. The data might not have column names.
You can use `col_names = FALSE` to tell `read_csv()` not to treat the first row as headings, and instead label them sequentially from `X1` to `Xn`:
```{r message = FALSE}
2016-07-07 02:59:50 +08:00
read_csv("1,2,3\n4,5,6", col_names = FALSE)
```
(`"\n"` is a convenient shortcut for adding a new line. You'll learn more about it and other types of string escape in Chapter \@ref(strings).)
Alternatively you can pass `col_names` a character vector which will be used as the column names:
```{r message = FALSE}
2016-07-07 02:59:50 +08:00
read_csv("1,2,3\n4,5,6", col_names = c("x", "y", "z"))
```
2015-09-23 02:35:39 +08:00
2016-07-27 03:57:25 +08:00
Another option that commonly needs tweaking is `na`: this specifies the value (or values) that are used to represent missing values in your file:
2016-07-12 04:38:39 +08:00
```{r message = FALSE}
2016-07-12 04:38:39 +08:00
read_csv("a,b,c\n1,2,.", na = ".")
```
This is all you need to know to read \~75% of CSV files that you'll encounter in practice.
You can also easily adapt what you've learned to read tab separated files with `read_tsv()` and fixed width files with `read_fwf()`.
To read in more challenging files, you'll need to learn more about how readr parses each column, turning them into R vectors.
### First steps
Let's take another look at the `students` data.
2022-03-05 12:58:23 +08:00
In the `favourite.food` column, there are a bunch of food items and then the character string `N/A`, which should have been an real `NA` that R will recognize as "not available".
This is something we can address using the `na` argument.
```{r message = FALSE}
students <- read_csv("data/students.csv", na = c("N/A", ""))
students
```
2022-03-05 12:58:23 +08:00
Once you read data in, the first step usually involves transforming it in some way to make it easier to work with in the rest of your analysis.
For example, the column names in the `students` file we read in are formatted in non-standard ways.
You might consider renaming them one by one with `dplyr::rename()` or you might use the `janitor::clean_names()` function turn them all into snake case at once.[^data-import-1]
This function takes in a data frame and returns a data frame with variable names converted to snake case.
2022-02-24 03:15:52 +08:00
[^data-import-1]: The [janitor](http://sfirke.github.io/janitor/) package is not part of the tidyverse, but it offers handy functions for data cleaning and works well within data pipelines that uses `|>`.
```{r message = FALSE}
library(janitor)
2022-02-24 03:15:52 +08:00
students |>
clean_names()
```
2022-03-05 12:58:23 +08:00
Another common task after reading in data is to consider variable types.
For example, `meal_type` is a categorical variable with a known set of possible values.
In R, factors can be used to work with categorical variables.
We can convert this variable to a factor using the `factor()` function.
You'll learn more about factors in Chapter \@ref(factors).
```{r}
2022-02-24 03:15:52 +08:00
students <- students |>
clean_names() |>
mutate(meal_plan = factor(meal_plan))
students
```
Note that the values in the `meal_type` variable has stayed exactly the same, but the type of variable denoted underneath the variable name has changed from character (`<chr>`) to factor (`<fct>`).
Before you move on to analyzing these data, you'll probably want to fix the `age` column as well: currently it's a character variable because of the one observation that is typed out as `five` instead of a numeric `5`.
We discuss the details of fixing this issue in Chapter \@ref(import-spreadsheets) in further detail.
2016-07-07 02:59:50 +08:00
### Compared to base R
2015-09-23 02:35:39 +08:00
If you've used R before, you might wonder why we're not using `read.csv()`.
2022-03-05 12:58:23 +08:00
There are a few good reasons to favor readr functions over the base equivalents:
2016-01-27 21:33:35 +08:00
- They are typically much faster (\~10x) than their base equivalents.
Long running jobs have a progress bar, so you can see what's happening.
If you're looking for raw speed, try `data.table::fread()`.
It doesn't fit quite so well into the tidyverse, but it can be quite a bit faster.
2016-01-27 21:33:35 +08:00
2022-03-05 12:58:23 +08:00
- They produce tibbles, and they don't use row names or munge the column names.
These are common sources of frustration with the base R functions.
2016-01-27 21:33:35 +08:00
- They are more reproducible.
2022-03-05 12:58:23 +08:00
Base R functions inherit some behavior from your operating system and environment variables, so import code that works on your computer might not work on someone else's.
2016-01-27 21:33:35 +08:00
2016-07-12 09:29:17 +08:00
### Exercises
2016-07-09 05:23:19 +08:00
2022-03-05 12:58:23 +08:00
1. What function would you use to read a file where fields were separated with "\|"?
2. Apart from `file`, `skip`, and `comment`, what other arguments do `read_csv()` and `read_tsv()` have in common?
3. What are the most important arguments to `read_fwf()`?
4. Sometimes strings in a CSV file contain commas.
To prevent them from causing problems they need to be surrounded by a quoting character, like `"` or `'`. By default, `read_csv()` assumes that the quoting character will be `"`.
What argument to `read_csv()` do you need to specify to read the following text into a data frame?
2016-07-14 22:05:20 +08:00
```{r, eval = FALSE}
2016-07-09 05:23:19 +08:00
"x,y\n1,'a,b'"
```
5. Identify what is wrong with each of the following inline CSV files.
2016-07-12 04:38:39 +08:00
What happens when you run the code?
2016-07-12 04:38:39 +08:00
```{r, eval = FALSE}
read_csv("a,b\n1,2,3\n4,5,6")
read_csv("a,b,c\n1,2\n1,2,3,4")
read_csv("a,b\n\"1")
read_csv("a,b\n1,2\na,b")
read_csv("a;b\n1;3")
```
2016-07-09 05:23:19 +08:00
## Reading data from multiple files
2016-07-08 00:17:11 +08:00
Sometimes your data is split across multiple files instead of being contained in a single file.
For example, you might have sales data for multiple months, with each month's data in a separate file: `01-sales.csv` for January, `02-sales.csv` for February, and `03-sales.csv` for March.
With `read_csv()` you can read these data in at once and stack them on top of each other in a single data frame.
2015-09-23 21:58:16 +08:00
2016-07-07 02:59:50 +08:00
```{r}
sales_files <- c("data/01-sales.csv", "data/02-sales.csv", "data/03-sales.csv")
read_csv(sales_files, id = "file")
2016-07-07 02:59:50 +08:00
```
2015-09-23 21:58:16 +08:00
With the additional `id` parameter we have added a new column called `file` to the resulting data frame that identifies the file the data come from.
This is especially helpful in circumstances where the files you're reading in do not have an identifying column that can help you trace the observations back to their original sources.
2016-07-07 02:59:50 +08:00
If you have many files you want to read in, it can get cumbersome to write out their names as a list.
2022-03-05 12:58:23 +08:00
Instead, you can use the `dir_ls()` function from the [fs](https://fs.r-lib.org/) package to find the files for you by matching a pattern in the file names.
2016-07-07 02:59:50 +08:00
2016-07-10 22:19:56 +08:00
```{r}
library(fs)
sales_files <- dir_ls("data", glob = "*sales.csv")
sales_files
2016-07-07 02:59:50 +08:00
```
2016-07-10 22:19:56 +08:00
## Writing to a file
2015-09-23 21:58:16 +08:00
readr also comes with two useful functions for writing data back to disk: `write_csv()` and `write_tsv()`.
Both functions increase the chances of the output file being read back in correctly by:
- Always encoding strings in UTF-8.
2016-07-07 02:59:50 +08:00
- Saving dates and date-times in ISO8601 format so they are easily parsed elsewhere.
2016-07-07 02:59:50 +08:00
2016-08-12 21:09:18 +08:00
If you want to export a csv file to Excel, use `write_excel_csv()` --- this writes a special character (a "byte order mark") at the start of the file which tells Excel that you're using the UTF-8 encoding.
The most important arguments are `x` (the data frame to save), and `file` (the location to save it).
You can also specify how missing values are written with `na`, and if you want to `append` to an existing file.
2016-07-07 02:59:50 +08:00
2016-07-10 22:19:56 +08:00
```{r, eval = FALSE}
write_csv(students, "students.csv")
2016-07-10 22:19:56 +08:00
```
2016-07-07 02:59:50 +08:00
2022-03-05 12:58:23 +08:00
Now let's read that csv file back in.
2016-07-10 22:19:56 +08:00
Note that the type information is lost when you save to csv:
2016-07-07 02:59:50 +08:00
```{r, warning = FALSE, message = FALSE}
students
write_csv(students, "students-2.csv")
read_csv("students-2.csv")
2016-07-07 02:59:50 +08:00
```
2015-09-23 21:58:16 +08:00
This makes CSVs a little unreliable for caching interim results---you need to recreate the column specification every time you load in.
There are two alternatives:
1. `write_rds()` and `read_rds()` are uniform wrappers around the base functions `readRDS()` and `saveRDS()`.
These store data in R's custom binary format called RDS:
2015-09-23 21:58:16 +08:00
2016-07-10 22:19:56 +08:00
```{r}
write_rds(students, "students.rds")
read_rds("students.rds")
2016-07-10 22:19:56 +08:00
```
2. The feather package implements a fast binary file format that can be shared across programming languages:
```{r, eval = FALSE}
2016-07-10 22:19:56 +08:00
library(feather)
write_feather(students, "students.feather")
read_feather("students.feather")
#> # A tibble: 6 × 5
#> student_id full_name favourite_food meal_plan age
#> <dbl> <chr> <chr> <fct> <dbl>
#> 1 1 Sunil Huffmann Strawberry yoghurt Lunch only 4
#> 2 2 Barclay Lynn French fries Lunch only 5
#> 3 3 Jayendra Lyne NA Breakfast and lunch 7
#> 4 4 Leon Rossini Anchovies Lunch only NA
#> 5 5 Chidiegwu Dunkel Pizza Breakfast and lunch 5
#> 6 6 Güvenç Attila Ice cream Lunch only 6
2016-07-10 22:19:56 +08:00
```
2015-09-21 21:41:14 +08:00
Feather tends to be faster than RDS and is usable outside of R.
2021-02-22 21:34:21 +08:00
RDS supports list-columns (which you'll learn about in Chapter \@ref(list-columns); feather currently does not.
2015-09-21 21:41:14 +08:00
2016-07-10 22:19:56 +08:00
```{r, include = FALSE}
file.remove("students-2.csv")
file.remove("students.rds")
2016-07-10 22:19:56 +08:00
```
2015-09-21 21:41:14 +08:00
2016-07-07 02:59:50 +08:00
## Other types of data
2016-04-02 01:33:51 +08:00
To get other types of data into R, we recommend starting with the tidyverse packages listed below.
They're certainly not perfect, but they are a good place to start.
For rectangular data:
2016-04-02 01:33:51 +08:00
- **readxl** reads Excel files (both `.xls` and `.xlsx`).
See Chapter \@ref(import-spreadsheets) for more on working with data stored in Excel spreadsheets.
2016-04-02 01:33:51 +08:00
- **googlesheets4** reads Google Sheets.
Also see Chapter \@ref(import-spreadsheets) for more on working with data stored in Google Sheets.
2016-04-02 01:33:51 +08:00
- **DBI**, along with a database specific backend (e.g. **RMySQL**, **RSQLite**, **RPostgreSQL** etc) allows you to run SQL queries against a database and return a data frame.
See Chapter \@ref(import-databases) for more on working with databases .
- **haven** reads SPSS, Stata, and SAS files.
For hierarchical data: use **jsonlite** (by Jeroen Ooms) for json, and **xml2** for XML.
Jenny Bryan has some excellent worked examples at <https://jennybc.github.io/purrr-tutorial/>.
2016-07-12 04:38:39 +08:00
For other file types, try the [R data import/export manual](https://cran.r-project.org/doc/manuals/r-release/R-data.html) and the [**rio**](https://github.com/leeper/rio) package.