Find And Update Array In Python