Pero "que problema tienes"?
Algún error? .. que hace tu rutina que no deba hacer o que no hace? .. explicate un poco méjor ..
En principio veo que te faltan (por seguridad) unas comillas en:
mysql_query("delete from prueba where codigo = $id",$conexion);
debería ser:
mysql_query("delete from prueba where codigo = '$id'",$conexion);
Y si quieres actualizar ese registro en función de esa condición "codigo=$id" .. tan sólo tienes que hacer un UPDATE (usando dicha sentencia SQL) . si no conoces como usarla:
www.mysql.com
Un saludo,