Insert values of dataframe to class atributes with for loop in python

Question

I have parsed data in dataframes by using Pandas. I need to insert the data into class that I created.

My class:

class Message:
time = None
id = None
type = None
source = None
destination = None
    def __init__(self, time, id, type, source, destination):
    self.time = time
    self.id = id
    self.type = type
    self.source = source
    self.destination = destination

I'm going through the dataframes and trying to insert the output into the attributes of the class as following:

newMessage=Message()
for index, row in df.iterrows():
    newMessage.__init__(row['time'], row['id'], row['type'], row['source'], row['destination'])
    print(row['time'], row['ID'], row['TYPE'], row['Source'], row['Destination'])

The exception it throws:

TypeError: __init__() missing 7 required positional arguments

I don't know how to call the class and the arguments, please help.

Before the for-loop, newMessage=Message() is trying to create a Message object, but you're not passing in any of the required arguments such as time, id, etc. It can't know what you want those values to be, so it gives an error. I'd suggest working through a python classes tutorial like w3schools.com/python/python_classes.asp — mitoRibo
– mitoRibo, Commented Aug 25, 2021 at 16:53

Marcel · Accepted Answer · 2021-08-25 17:57:06Z

1

You could try something like that:

class Message:

    time = None
    id_ = None
    type_ = None
    source = None
    destination = None


def __init__(self, time, id_, type_, source, destination):
    self.time = time
    self.id_ = id_
    self.type_ = type_
    self.source = source
    self.destination = destination

for index, row in df.iterrows():
    newTelegram = Message(row['time'], row['id'], row['type'], row['source'], row['destination'])
 
 print(newTelegram.time)
 print(getattr(newTelegram, "time"))

Note that both id & type are keywords in Python.

answered Aug 25, 2021 at 17:57

Marcel

4844 silver badges9 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

vloubes Over a year ago

I think that is what I needed. Thank you!

Ade_1 · Accepted Answer · 2021-10-03 01:53:47Z

1

You can explore the powerful use of dataclasses or NamedTuple:

from dataclasses import dataclass

@dataclass
class Message:
    time: str    #adjust the datatype to suit your usecase
    id:  int
    type_: str
    source:  str
    destination: str 

for index, value in df.iterrows():
    newTelegram = Message(*value)# unpack the values

edited Oct 3, 2021 at 1:53

answered Aug 25, 2021 at 18:15

Ade_1

1,5161 gold badge8 silver badges17 bronze badges

3 Comments

vloubes Over a year ago

That one looks good too. Thanks for your reply!

ListenSoftware Louise Ai Agent Over a year ago

did you mean *value

Ade_1 Over a year ago

oh yeah, my bad. thanks

balderman · Accepted Answer · 2021-08-25 18:13:55Z

0

How about the below ? Use dataclass and get a clean solution

import pandas as pd
from dataclasses import dataclass

@dataclass
class Point:
  x: int
  y: int

points = []
df = pd.DataFrame([{'x':12,'y':6},{'x':45,'y':33}])
for index, row in df.iterrows():
  points.append(Point(**row))
print(points)

output

[Point(x=12, y=6), Point(x=45, y=33)]

answered Aug 25, 2021 at 18:13

balderman

24k8 gold badges39 silver badges60 bronze badges

Collectives™ on Stack Overflow

Insert values of dataframe to class atributes with for loop in python

3 Answers 3

1 Comment

3 Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

1 Comment

3 Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related