Python TypeError: 'newline' is an invalid keyword argument for this function

最新推荐文章于 2022-07-15 11:22:54 发布

原创最新推荐文章于 2022-07-15 11:22:54 发布 · 8.2k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#Python

DataScience 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一段用于将数据写入CSV文件的Python代码，并解决了在不同Python版本中出现的newline参数问题。通过版本判断，代码能够在Python2和Python3下正确运行，确保了跨版本的兼容性。

写一个存为csv文件的代码

 with open(outputFile, 'w', newline='') as csvfile:
     writer = csv.writer(csvfile)
     for item in sortsim:
         writer.writerow([item[0], item[1], item[2]])

出现问题了

Traceback (most recent call last):
  File "/data/ml/shan-als.py", line 54, in <module>
    with open(outputFile, 'w', newline='') as csvfile:
TypeError: 'newline' is an invalid keyword argument for this function

猜测是版本的原因，加了版本判断。
py2用’wb’，py3可以用newline=’’。

    import sys
    if sys.version >= '3':
        with open(outputFile, 'w', newline='') as csvfile:
            writer = csv.writer(csvfile)
            for item in sortsim:
                writer.writerow([item[0], item[1], item[2]])
    else:
        with open(outputFile, 'wb') as csvfile:
            writer = csv.writer(csvfile)
            for item in sortsim:
                writer.writerow([item[0], item[1], item[2]])