Python Get Dimension Of Array