使用循环将CSV数据提取到obj中

2024-04-29 13:46:42 发布

您现在位置:Python中文网/ 问答频道 /正文

我的问题很难解释,我不确定是否有可能做到我所要求的,但我会尽力解释。你知道吗

基本上,我有一个包含数据的CSV文件,我想提取特定的单元格,并将它们设置为对象中的值。CSV中的每一行都包含有关单个项目的信息。目前,我使用熊猫库和测向iloc[0][1]等。但是,我希望能够循环浏览整个CSV,提取单个单元格并将它们添加到多个对象中,这样就不必手动硬编码每一行。你知道吗

希望代码能帮助你明白我的意思:

df = pd.read_csv('Options.csv')

我的班级:

    class Option:
    def __init__(self, type, name, S, K):

        self.type = type
        self.name = name
        self.S = S
        self.K = K 

当前从CSV提取:

o1 = Option(df.iloc[0, 1], df.iloc[0][2], df.iloc[0][3], df.iloc[0][4])
o2 = Option(df.iloc[1, 1], df.iloc[1][2], df.iloc[1][3], df.iloc[1][4])

等等

我仍然希望能够选择每个选项的单个值,例如print(o1.name)、o6.type等


Tags: 文件csv数据项目对象nameself信息
2条回答

您需要迭代文档中从0到最多行的每一行。对于每个迭代,您将创建Option类的一个新实例。你知道吗

下面是一些示例代码:

df = pd.read_csv('Option.csv')

row_amount = 5 # put the amount of rows that the csv file contains here or get it with pandas 

options = []

for row in range(0, row_amount):
    option = Option(df.iloc[row, 1], df.iloc[row, 2], df.iloc[row, 3], df.iloc[row, 4])
    options.append(option)

这将为您提供选项对象的列表:

options = df.apply(lambda x: Option(x[1], x[2], x[3], x[4]), axis=1)

options_list = options.values.tolist()

相关问题 更多 >