Отображение содержимого массива в UITextView на основе выбора строки в Swift

В plt.colorbar(z1_plot,cax=ax1) используйте ax= вместо cax=, то есть plt.colorbar(z1_plot,ax=ax1)