| <!--- |
| Licensed to the Apache Software Foundation (ASF) under one |
| or more contributor license agreements. See the NOTICE file |
| distributed with this work for additional information |
| regarding copyright ownership. The ASF licenses this file |
| to you under the Apache License, Version 2.0 (the |
| "License"); you may not use this file except in compliance |
| with the License. You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, |
| software distributed under the License is distributed on an |
| "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| KIND, either express or implied. See the License for the |
| specific language governing permissions and limitations |
| under the License. |
| --> |
| |
| (io_csv)= |
| |
| # CSV |
| |
| Reading a csv is very straightforward with {py:func}`~datafusion.context.SessionContext.read_csv` |
| |
| ```python |
| from datafusion import SessionContext |
| |
| ctx = SessionContext() |
| df = ctx.read_csv("file.csv") |
| ``` |
| |
| An alternative is to use {py:func}`~datafusion.context.SessionContext.register_csv` |
| |
| ```python |
| ctx.register_csv("file", "file.csv") |
| df = ctx.table("file") |
| ``` |
| |
| If you require additional control over how to read the CSV file, you can use |
| {py:class}`~datafusion.options.CsvReadOptions` to set a variety of options. |
| |
| ```python |
| from datafusion import CsvReadOptions |
| options = ( |
| CsvReadOptions() |
| .with_has_header(True) # File contains a header row |
| .with_delimiter(";") # Use ; as the delimiter instead of , |
| .with_comment("#") # Skip lines starting with # |
| .with_escape("\\") # Escape character |
| .with_null_regex(r"^(null|NULL|N/A)$") # Treat these as NULL |
| .with_truncated_rows(True) # Allow rows to have incomplete columns |
| .with_file_compression_type("gzip") # Read gzipped CSV |
| .with_file_extension(".gz") # File extension other than .csv |
| ) |
| df = ctx.read_csv("data.csv.gz", options=options) |
| ``` |
| |
| Details for all CSV reading options can be found on the |
| [DataFusion documentation site](https://datafusion.apache.org/library-user-guide/custom-table-providers.html). |