Writing To Array Python