swap_cols
- pyhelpers.ops.swap_cols(array, c1, c2, as_list=False)
Swap positions of two columns in an array.
- Parameters:
array (numpy.ndarray) – an array
c1 (int) – index of a column
c2 (int) – index of another column
as_list (bool) – whether to return a list
- Returns:
a new array/list in which the positions of the c1-th and c2-th columns are swapped
- Return type:
numpy.ndarray or list
Examples:
>>> from pyhelpers.ops import swap_cols >>> from pyhelpers._cache import example_dataframe >>> example_arr = example_dataframe(osgb36=True).to_numpy(dtype=int) >>> example_arr array([[530039, 180371], [406705, 286868], [383830, 398113], [430147, 433553]]) >>> # Swap the 0th and 1st columns >>> new_arr = swap_cols(example_arr, c1=0, c2=1) >>> new_arr array([[180371, 530039], [286868, 406705], [398113, 383830], [433553, 430147]]) >>> new_list = swap_cols(example_arr, c1=0, c2=1, as_list=True) >>> new_list [[180371, 530039], [286868, 406705], [398113, 383830], [433553, 430147]]