series数据操作，float转int(四舍五入)

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/Perfect_Accepted/article/details/81779083

import pandas as pd

y = pd.read_csv('./0.09.csv')  # y为DataFrame格式

# 将float转换为int，四舍五入
# 这里y['class']为series格式，直接对其操作显示无法操作，需将其里面的数单独取出来操作，这里取出来的数为float格式
y_int = []
for num in y['class']:
    y_float = (num * 10 + 5) / 10
    y_int.append(int(y_float))

# 将数据写入csv文件

file = "./y_pred_int.csv"
index_num = 0

with open(file, 'w') as f:
    for item in y_int:
        f.write(str(index_num) + "," + str(item) + "\n")
        index_num += 1