Python из памяти большого CSV файла (numpy)

У меня есть CSV файл размером 3 ГБ, который я пытаюсь читать с помощью python, мне нужен средний столбец.

from numpy import * 
def data():
    return genfromtxt('All.csv',delimiter=',')

data = data() # This is where it fails already.

med = zeros(len(data[0]))
data = data.T
for i in xrange(len(data)):
    m = median(data[i])
    med[i] = 1.0/float(m)
print med

Ошибка, которую я получаю, такова:

Python(1545) malloc: *** mmap(size=16777216) failed (error code=12)

*** error: can't allocate region

*** set a breakpoint in malloc_error_break to debug

Traceback (most recent call last):

  File "Normalize.py", line 40, in <module>

  data = data()

  File "Normalize.py", line 39, in data

  return genfromtxt('All.csv',delimiter=',')

File "/Library/Frameworks/Python.framework/Versions/2.6/lib/python2.6/site-
packages/numpy/lib/npyio.py", line 1495, in genfromtxt

for (i, line) in enumerate(itertools.chain([first_line, ], fhd)):

MemoryError

Я думаю, что это просто ошибка из памяти. Я запускаю 64-битный MacOSX с 4 ГБ оперативной памяти, и как numpy, так и Python, скомпилированные в 64-битном режиме.

Как это исправить? Должен ли я попробовать распределенный подход, только для управления памятью?

Спасибо

EDIT: Также пробовал с этим, но не повезло...

genfromtxt('All.csv',delimiter=',', dtype=float16)

Ответ 1

Как отмечали другие люди, для действительно большого файла вам лучше выполнять итерацию.

Однако, вы часто хотите, чтобы все было в памяти по разным причинам.

genfromtxt намного менее эффективен, чем loadtxt (хотя он обрабатывает отсутствующие данные, тогда как loadtxt более "тощий и средний", поэтому две функции сосуществуют).

Если ваши данные очень регулярные (например, просто простые строки с одинаковым типом), вы также можете улучшить либо с помощью numpy.fromiter.

Если у вас достаточно бара, подумайте об использовании np.loadtxt('yourfile.txt', delimiter=',') (вам также может потребоваться указать skiprows, если у вас есть заголовок в файле.)

В качестве быстрого сравнения загружаемый ~ 500 МБ текстовый файл с loadtxt использует ~ 900 МБ оперативной памяти при максимальном использовании, а при загрузке того же файла с genfromtxt используется ~ 2,5 ГБ.

Loadtxt Memory and CPU usage of numpy.loadtxt while loading a ~500MB ascii file


Genfromtxt Memory and CPU usage of numpy.genfromtxt while loading a ~500MB ascii file


В качестве альтернативы рассмотрите следующее. Он будет работать только для очень простых, регулярных данных, но это довольно быстро. (loadtxt и genfromtxt делать много угадывания и проверки ошибок. Если ваши данные очень простые и регулярные, вы можете улучшить их значительно.)

import numpy as np

def generate_text_file(length=1e6, ncols=20):
    data = np.random.random((length, ncols))
    np.savetxt('large_text_file.csv', data, delimiter=',')

def iter_loadtxt(filename, delimiter=',', skiprows=0, dtype=float):
    def iter_func():
        with open(filename, 'r') as infile:
            for _ in range(skiprows):
                next(infile)
            for line in infile:
                line = line.rstrip().split(delimiter)
                for item in line:
                    yield dtype(item)
        iter_loadtxt.rowlength = len(line)

    data = np.fromiter(iter_func(), dtype=dtype)
    data = data.reshape((-1, iter_loadtxt.rowlength))
    return data

#generate_text_file()
data = iter_loadtxt('large_text_file.csv')

Fromiter

Using fromiter to load the same ~500MB data file

Ответ 2

Проблема с использованием genfromtxt() заключается в том, что он пытается загрузить весь файл в память, т.е. в массив numpy. Это отлично подходит для небольших файлов, но BAD для 3 ГБ входов, подобных вашим. Поскольку вы просто вычисляете медианы столбцов, нет необходимости читать весь файл. Простым, но не самым эффективным способом сделать это было бы чтение целого файла по очереди несколько раз и повторение по столбцам.

Ответ 3

Почему вы не используете модуль python csv?

>> import csv
>> reader = csv.reader(open('All.csv'))
>>> for row in reader:
...     print row