Python pandas: выводит фрейм данных в csv с целыми числами
у меня естьpandas.DataFrame
что я хочу экспортировать в файл CSV. Тем не менее, панды, кажется, пишут некоторые значения какfloat
вместоint
типы. Я не могНе могу найти, как изменить это поведение.
Создание фрейма данных:
df = pandas.DataFrame(columns=['a','b','c','d'], index=['x','y','z'], dtype=int)
x = pandas.Series([10,10,10], index=['a','b','d'], dtype=int)
y = pandas.Series([1,5,2,3], index=['a','b','c','d'], dtype=int)
z = pandas.Series([1,2,3,4], index=['a','b','c','d'], dtype=int)
df.loc['x']=x; df.loc['y']=y; df.loc['z']=z
Посмотреть это:
>>> df
a b c d
x 10 10 NaN 10
y 1 5 2 3
z 1 2 3 4
Экспортируйте это:
>>> df.to_csv('test.csv', sep='\t', na_rep='0', dtype=int)
>>> for l in open('test.csv'): print l.strip('\n')
a b c d
x 10.0 10.0 0 10.0
y 1 5 2 3
z 1 2 3 4
Почему десятки имеют точку ноль?
Конечно, я мог бы просто вставить эту функцию в свой конвейер, чтобы преобразовать весь файл CSV, но это кажется ненужным:
def lines_as_integer(path):
handle = open(path)
yield handle.next()
for line in handle:
line = line.split()
label = line[0]
values = map(float, line[1:])
values = map(int, values)
yield label + '\t' + '\t'.join(map(str,values)) + '\n'
handle = open(path_table_int, 'w')
handle.writelines(lines_as_integer(path_table_float))
handle.close()