numpy.savetxt приводит к ошибке несоответствия форматирования в python 3.5

Я пытаюсь сохранить матрицу numpy (Nx3, float64) в текстовый файл, используя numpy.savetxt:

np.savetxt(f, mat, fmt='%.5f', delimiter=' ')

Эта строка работала в python 2.7, но в python 3.5 я получаю следующую ошибку:

TypeError: несоответствие между dtype массива ("float64") и спецификатором формата ("%.5f %.5f %.5f")

Когда я вхожу в код savetxt, печатаю ошибку (traceback.format_exc()) в блоке catch (numpy.lib.npyio, строка 1158), ошибка совершенно другая:

TypeError: аргумент write() должен быть строкой, а не байтами

Строка кода, приводящая к исключению, выглядит следующим образом:

fh.write(asbytes(format % tuple(row) + newline))

Я попытался удалить асбайты, и это, кажется, исправило эту ошибку. Это ошибка в numpy?


numpy formatting python-3.5
person Noam Peled    schedule 12.02.2016    source источник

Ответы (1)


arrow_upward
23
arrow_downward

savetxt открывает файл в режиме wb и, таким образом, записывает все как байты.

Если вместо этого я открою файл с помощью «w», я получу вторую ошибку:

In [403]: x=np.ones((3,3),dtype=np.float64)
In [404]: with open('test.txt','w') as f:
    np.savetxt(f,x,fmt='%.5f')
   .....:     
TypeError: must be str, not bytes

Но нет проблем с

In [405]: with open('test.txt','wb') as f:
    np.savetxt(f,x,fmt='%.5f')
   .....:     
In [406]: cat test.txt
1.00000 1.00000 1.00000
1.00000 1.00000 1.00000
1.00000 1.00000 1.00000

Это на Py3.4; У меня не установлено numpy с моим 3.5 Python. Но я бы не ожидал разницы.

Делает

'%.5f'%1.2342

работать в вашей системе? Вы также можете попробовать

'%.5f %.5f %.5f'%tuple(mat[0,:])
person hpaulj    schedule 12.02.2016
comment
Спасибо! Это сработало! Я также пишу текст в этот файл, поэтому сначала мне нужно было открыть его с помощью «w», а затем снова открыть с помощью «ab». - person Noam Peled; 13.02.2016
comment
Вы можете создавать байтовые строки с помощью b'one two three'. - person hpaulj; 13.02.2016
comment
У меня была такая же ошибка при записи объекта io.StringIO. В этом случае решение состоит в том, чтобы заменить его объектом io.BytesIO. - person SiggyF; 05.07.2017
comment
Вау, это помогло с другой проблемой, которая у меня была, хотя я добавлял файл file=open(file, 'ab') mode, и он работает с текстом сохранения (только что добавил b) - person Matt; 06.07.2018