Python Sort 2d Array By First Element With Atomic Mass