How can I add custom methods to a dataclass in Python?

How can I add custom methods to a dataclass in Python?

You can add custom methods to a dataclass just like you would in a regular class. Define the method inside the class body.

Can you give me an example of a dataclass with a custom method?

Sure! Here's a simple example:

from dataclasses import dataclass

@dataclass
class Point:
    x: int
    y: int

    def distance_from_origin(self):
        return (self.x**2 + self.y**2) ** 0.5

This Point class has a method to calculate the distance from the origin.

What if I want to add a method that modifies the dataclass attributes?

You can definitely do that! Just define a method that modifies the attributes directly. For example:

def move(self, dx, dy):
    self.x += dx
    self.y += dy

Thanks! Can I also use properties in a dataclass?

Yes, you can use properties in a dataclass. Just use the @property decorator to define a method as a property.