python 怎样把一个数组类型数据保存为csv文件

有时候做完一个案例的分析,模型做好了,最后想把预测的结果保存为CSV文件。这个预测的结果一般是n行1列的一个数列,我们最后就是想把预测结果保存下来。

submission = clf.predict(credit_test)#submission是我的预测结果,是一个5000*1的列向量

submission = pd.DataFrame({'RISK':submission})

#必须把这个submission用pd.DataFrame读出来

不然会报错 AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'

submission.to_csv('submission.csv',index=False)


以后遇到“AttributeError: 'numpy.ndarray' object has no attribute 'to_csv'”就可以用上面的方法解决

也可以用这个submission=pd.Series(submission)替换submission = pd.DataFrame({'RISK':submission})

  • 11
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 好的,以下是可以回答的问题: 可以的,以下是一个读取csv文件保存arrayPython函数: ```python import csv import numpy as np def read_csv_file(file_path): with open(file_path, 'r') as f: reader = csv.reader(f) data = np.array(list(reader)) return data ``` 你可以将文件路径作为参数传递给这个函数,它将返回一个包含csv文件数据numpy数组。 请注意,这个函数只适用于简单的csv文件,如果文件包含复杂的数据类型或格式,则需要进行适当的修改。 还有其他问题吗? ### 回答2: 当人们处理csv文件时,通常使用pandas库,因为它提供了许多方便的函数和方法来处理和操作数据。下面是一个使用pandas库的Python函数,可以读取csv文件并将其保存数组: ```python import pandas as pd import numpy as np def read_csv_file(file_path): data_df = pd.read_csv(file_path) # 读取csv文件并将其存储在DataFrame对象中 data_array = np.array(data_df) # 将DataFrame对象转换为数组 return data_array ``` 这个函数需要一个参数`file_path`,表示要读取的csv文件的路径。函数首先使用`pd.read_csv()`函数来读取csv文件并将其存储在DataFrame对象`data_df`中。然后,使用`np.array()`函数将DataFrame对象转换为Numpy数组`data_array`。最后,将数组返回。 你可以调用这个函数并传递你的csv文件路径作为参数,接收返回的数组。例如: ```python file_path = 'path/to/your/csv/file.csv' # 替换为你的csv文件的实际路径 data = read_csv_file(file_path) print(data) ``` 在控制台中,你将看到打印出的数组数据。请确保在使用这个函数之前安装了pandas和numpy库。你可以使用以下命令来安装它们: ``` pip install pandas pip install numpy ``` 希望这可以帮助你读取csv文件并将其保存数组。如果有进一步的问题,请随时提问。 ### 回答3: 可以使用Pythoncsv模块来实现这个函数。具体的代码如下: ```python import csv import numpy as np def read_csv(file_path): data = [] # 用于保存csv文件中的数据 # 打开csv文件并读取数据 with open(file_path, 'r') as file: reader = csv.reader(file) for row in reader: data.append(row) # 将数据转换为numpy数组 array_data = np.array(data) return array_data ``` 这个函数接受一个参数`file_path`,表示要读取的csv文件的路径。函数会打开csv文件,使用csv.reader逐行读取数据,并将每一行数据添加到`data`列表中。然后,使用numpy的`array`函数将`data`转换为numpy数组`array_data`。最后,函数返回这个numpy数组。 使用这个函数可以轻松地读取csv文件中的数据,并保存一个array。例如,假设csv文件的路径是`data.csv`,可以通过以下代码调用这个函数: ```python data_array = read_csv('data.csv') ``` 这样,`data_array`就存储了csv文件中的数据,并可以使用numpy提供的丰富功能对数据进行操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值