I have a code to find the electric field and the potential. i want to re write it as class but in easier way

The code:

class ChargedParticle: def __init__(self, pos, charge): self.pos = np.asarray(pos) self.charge = charge def compute_field(self, x, y): # this part i want to rewrite it X, Y = np.meshgrid(x, y) u_i = np.hstack((X.ravel()[:, np.newaxis], Y.ravel()[:, np.newaxis])) - self.pos r = np.sqrt((X - self.pos[0]) ** 2 + (Y - self.pos[1]) ** 2) field = ((self.charge / r ** 2).ravel()[:, np.newaxis] * u_i).reshape(X.shape + (2,)) return field def compute_potential(self, x, y): X, Y = np.meshgrid(x, y) r = np.sqrt((X - self.pos[0]) ** 2 + (Y - self.pos[1]) ** 2) potential = self.charge / r return potential