Python：将嵌套列表对象写入 csv 文件

Question

我正在尝试将列表列表中的数据写入 csv 文件。 这是我所拥有的简化版本

class Point(object): 
    def __init__(self, weight, height):
        self.weight = weight
        self.height = height
    def get_BMI(self):
        return (self.weight * self.height) / 42  # this is not how you calculate BMI but let's say

myList = [[Point(30, 183)],[Point(63, 153)]]

由于数据的设置方式，我将点存储在嵌套循环中。 如果我想访问第一个点对象的 BMI，我会输入

myList[0][0].get_BMI()

我想将每个点的 BMI 写入 CSV（以逗号分隔）。 我怎么做？

这是我的想法，但并不完全是直截了当的：

import csv
with open('output.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(myList)

它不会返回任何错误，但实际上也不会创建 CSV 文件。 另外我想将myList[i][j].get_BMI()的值写入文件。 我没有权限问题，因为我使用 Spyder（python IDE）作为 root。 现在我只是通过 Spyder 控制台运行脚本，但它应该仍然可以工作并输出 CSV 文件。

Answer 1

writerows需要一个字符串或数字列表。 您应该首先创建一个包含 BMI 值的列表，以便将它们写入您的csv文件：

import csv
BMIs = [[point.get_BMI() for point in points] for points in myList]
with open('output.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(BMIs)

Answer 2

存在三个问题：

1. 嵌套列表必须展平。 为此，请使用itertools.chain.from_iterable 。

2. 必须自定义 CSV 的行数据。 要做到这一点，请使用列表理解。

3. 未创建 output.csv。 我怀疑正在创建 output.csv 但被放置在意外的位置。 您可以尝试对完整路径进行硬编码以进行测试，以查看是否是这种情况。

这是演示 #1 和 #2 的代码：

import csv
from itertools import chain

with open('output.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    flattenedList = chain.from_iterable(myList)

    writer.writerows((pt.weight, pt.height, pt.get_BMI()) for pt in flattenedList)

Answer 3

您可以使用writerows来做到这writerows ，因为它需要一个rows列表 - 每row应该按照csv.writer的dialect参数进行csv.writer ，在这种情况下可以忽略它而不必担心任何适得其反。

所以writerows可以采用类似于myList的结构。 问题是您需要访问所有点并获取它们的 BMI（这可以在列表理解中完成）

为了说明如何使用 writerows（并为每个点添加一个数字，以便您的所有行都不会只有一个条目（这会令人沮丧），我添加了对 enumerate 的调用。

因此，您不再需要复杂的循环或任何此类。 享受：

myList = [[Point(30, 183)],[Point(63, 153)]]

with open('output.csv', 'w') as outfile:
    writer = csv.writer(outfile)
    writer.writerows(enumerate(p.get_BMI() for p in itertools.chain.from_iterable(myList)))

Answer 4

myList = [[Point(30, 183)],[Point(63, 153)]]

with open('output.csv', 'w') as outfile:
    writer = csv.writer(outfile,delimiter=',')
    for i in myList:
         writer.writerow([i])