从数据框加载对象答案

【问题标题】：Load object from Dataframe从数据框加载对象
【发布时间】：2018-01-16 14:28:24
【问题描述】：

我有一个对象基因

class Gene:
    def __init__(self, id, nb_obj, nb_days):
        self.id=id
        self.nb_obj=nb_obj
        self.nb_days=nb_days

我有一个熊猫数据框 df

   ID               nb_obj              nb_days
ECGYE                10259            62.965318
NLRTM                 8007            46.550562

如何将数据框的每一行加载为 Gene 对象？

【问题讨论】：

【解决方案1】：

你可以使用apply调用Gene的构造函数：

df.apply(lambda row: Gene(row['id'],row['nb_obj'],row['nb_days']), axis=1)

【讨论】：

【解决方案2】：

你可以试试这个：

[Gene(df.ID.loc[i], df.nb_obj.loc[i], df.nb_days.loc[i]) for i in range(df.shape[0])]

【讨论】：