Python Array Slice Elements From Atomic Number