绘制python对象属性和对象存储在列表中

Question

我在python中将对象存储在名为 lst 的列表中。我需要将一个对象属性绘制为绘图。

import numpy as np
import matplotlib.pyplot as plt

class Particle(object):
    def __init__(self, value = 0, weight = 0):
        self.value = value
        self.weight = weight
lst = []
for x in range(0,10):
    lst.append(Particle(value=np.random.random_integers(10), weight = 1))

我尝试了这个并且它有效，但我认为它不是非常'pythonic'方式：

temp = [] #any better idea? 
for x in range(0,len(lst)):
    temp.append(l[x].value)
plt.plot(temp, 'ro')

你有什么建议，如何用更加pythonic的方式来打它？ 谢谢

Answer 1

使用list comprehension生成值列表。

import numpy as np
import matplotlib.pyplot as plt

class Particle(object):
    def __init__(self, value = 0, weight = 0):
        self.value = value
        self.weight = weight
lst = []
for x in range(0,10):
    lst.append(Particle(value=np.random.random_integers(10), weight = 1))

values = [x.value for x in lst]

plt.plot(values, 'ro')
plt.show()

列表推导等同于以下代码：

values = []
for x in lst:
    values.append(x.value)

请注意，您还可以使用其他列表理解来整理lst集合的创建

lst = [(Particle(value=np.random.random_integers(10), weight=1) for _ in range(10)]