更新为使用 pandas 0.13.1
1) 编号http://pandas.pydata.org/pandas-docs/dev/io.html#notes-caveats。有多种方法可以执行此操作,例如让你的不同线程/进程写出计算结果,然后合并一个进程。
2) 根据您存储的数据类型、存储方式以及检索方式,HDF5 可以提供更好的性能。将 HDFStore 存储为单个数组,浮点数据,压缩(换句话说,不以允许查询的格式存储),存储/读取速度惊人。即使以表格格式存储(这会降低写入性能),也会提供相当好的写入性能。您可以查看一些详细的比较(这是 HDFStore 在后台使用的)。 http://www.pytables.org/,这是一张漂亮的照片:
(并且由于 PyTables 2.3 查询现在被索引),所以 perf 实际上比这要好得多
因此,回答您的问题,如果您想要任何类型的性能,HDF5 是您的最佳选择。
写作:
In [14]: %timeit test_sql_write(df)
1 loops, best of 3: 6.24 s per loop
In [15]: %timeit test_hdf_fixed_write(df)
1 loops, best of 3: 237 ms per loop
In [16]: %timeit test_hdf_table_write(df)
1 loops, best of 3: 901 ms per loop
In [17]: %timeit test_csv_write(df)
1 loops, best of 3: 3.44 s per loop
阅读
In [18]: %timeit test_sql_read()
1 loops, best of 3: 766 ms per loop
In [19]: %timeit test_hdf_fixed_read()
10 loops, best of 3: 19.1 ms per loop
In [20]: %timeit test_hdf_table_read()
10 loops, best of 3: 39 ms per loop
In [22]: %timeit test_csv_read()
1 loops, best of 3: 620 ms per loop
这是代码
import sqlite3
import os
from pandas.io import sql
In [3]: df = DataFrame(randn(1000000,2),columns=list('AB'))
<class 'pandas.core.frame.DataFrame'>
Int64Index: 1000000 entries, 0 to 999999
Data columns (total 2 columns):
A 1000000 non-null values
B 1000000 non-null values
dtypes: float64(2)
def test_sql_write(df):
if os.path.exists('test.sql'):
os.remove('test.sql')
sql_db = sqlite3.connect('test.sql')
sql.write_frame(df, name='test_table', con=sql_db)
sql_db.close()
def test_sql_read():
sql_db = sqlite3.connect('test.sql')
sql.read_frame("select * from test_table", sql_db)
sql_db.close()
def test_hdf_fixed_write(df):
df.to_hdf('test_fixed.hdf','test',mode='w')
def test_csv_read():
pd.read_csv('test.csv',index_col=0)
def test_csv_write(df):
df.to_csv('test.csv',mode='w')
def test_hdf_fixed_read():
pd.read_hdf('test_fixed.hdf','test')
def test_hdf_table_write(df):
df.to_hdf('test_table.hdf','test',format='table',mode='w')
def test_hdf_table_read():
pd.read_hdf('test_table.hdf','test')
当然是 YMMV。