Store Values In Array Python