remplazar una linea

def remplazar_linea(archivo, i, linea):
  with open(archivo, 'r') as file:
    lineas = file.readlines()
    lineas[i] = linea
  with open(archivo, 'w') as file:
    file.writelines(lineas)
 
remplazar_linea('archivo.txt', 2, 'linea a reemplazar\n')