我想使用類屬性來制作所有屬性的單個列表：from dataclasses import dataclass@dataclassclass Foo: a: int b: intdata = [ Foo(1,2), Foo(3,4)]print([[foo.a, foo.b] for foo in data])但結果是一列屬性列表。我怎樣才能有一個包含所有屬性的列表[1,2,3,4]而不是[[1,2],[3,4]]？
4 回答
眼眸繁星
TA貢獻1873條經驗 獲得超9個贊
我建議你使用itertools.chain.from_iterable來展平中間列表
In [6]: import itertools
In [7]: print(list(itertools.chain.from_iterable([[foo.a, foo.b] for foo in data])))
[1, 2, 3, 4]
慕田峪7331174
TA貢獻1828條經驗 獲得超13個贊
from functools import reduce
from dataclasses import dataclass
@dataclass
class Foo:
a: int
b: int
data = [
Foo(1,2),
Foo(3,4)
]
result = [[foo.a, foo.b] for foo in data]
print(reduce(list.__add__, result))
翻過高山走不出你
TA貢獻1875條經驗 獲得超3個贊
你可以這樣做：
print([value for foo in data for attr, value in foo.__dict__.items()])
它基本上是在做兩個 for 循環，與做的一樣：
for foo in data:
for attr, value in foo.__dict__.items():
print(value)
添加回答
舉報
0/150
提交
取消