欢迎访问宙启技术站
智能推送

pandas.DataFrame中的数据读取和写入方法有哪些

发布时间:2023-12-17 14:58:50

pandas库是一个流行的数据处理和分析工具,它提供了丰富的方法来读取和写入数据。下面是一些pandas.DataFrame中数据读取和写入的常用方法,以及它们的使用示例:

1. 读取CSV文件:使用pandas.read_csv()方法可以读取CSV格式的数据文件,并将其转换为DataFrame对象。

import pandas as pd

# 从CSV文件中读取数据,并创建DataFrame对象
df = pd.read_csv('data.csv')

2. 写入CSV文件:使用pandas.DataFrame.to_csv()方法可以将DataFrame对象中的数据写入到CSV文件中。

import pandas as pd

# 创建一个DataFrame对象
df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6]})

# 将DataFrame中的数据写入CSV文件
df.to_csv('data.csv', index=False)  # 不包含行索引

3. 读取Excel文件:使用pandas.read_excel()方法可以读取Excel文件,并将其转换为DataFrame对象。

import pandas as pd

# 从Excel文件中读取数据,并创建DataFrame对象
df = pd.read_excel('data.xlsx', sheet_name='Sheet1')

4. 写入Excel文件:使用pandas.DataFrame.to_excel()方法可以将DataFrame对象中的数据写入到Excel文件中。

import pandas as pd

# 创建一个DataFrame对象
df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6]})

# 将DataFrame中的数据写入Excel文件
df.to_excel('data.xlsx', index=False)  # 不包含行索引

5. 读取SQL数据库数据:使用pandas.read_sql()方法可以从SQL数据库中读取数据,并将其转换为DataFrame对象。需要先安装sqlalchemy库来连接数据库。

import pandas as pd
from sqlalchemy import create_engine

# 连接到SQL数据库
engine = create_engine('sqlite:///data.db')

# 从数据库中读取数据,并创建DataFrame对象
query = 'SELECT * FROM table_name'
df = pd.read_sql(query, engine)

6. 写入SQL数据库数据:使用pandas.DataFrame.to_sql()方法可以将DataFrame对象中的数据写入到SQL数据库中。

import pandas as pd
from sqlalchemy import create_engine

# 连接到SQL数据库
engine = create_engine('sqlite:///data.db')

# 创建一个DataFrame对象
df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6]})

# 将DataFrame中的数据写入数据库
df.to_sql('table_name', engine, if_exists='replace')

以上是一些常用的数据读取和写入方法,pandas还提供了很多其他的读取和写入数据的方法,可以根据具体的需求选择合适的方法。