Как исключить данные на основе 3-го параметра?

Question

У меня есть график зависимости радиусов pl anet от их орбитального периода, цвета, обозначенного их инсоляцией.

Как видно из графика, Значение инсоляции достигает 40000 раз больше, чем у Земли. Меня не волнуют эти ценности. Я забочусь в основном о значениях от 0 до 1,50. Есть ли способ, которым я могу исключить большие данные или, по крайней мере, сделать все, что выше 1,50, другим цветом? Пока у меня есть это:

plt.figure(figsize = (9,7))
plt.title('Planet Radius vs Orbital Period', fontsize = 24, fontweight='bold')
plt.ylabel('Radius [Earth Radii]', fontsize = 16)
plt.xlabel('Period [Days]', fontsize=16)
plt.xscale('log')
plt.yscale('log')

x = data2['pl_orbper']
y = data2['pl_rade']
z = data2['pl_insol']
plt.scatter(x,y, c=z, s=5, cmap='autumn')
cbar= plt.colorbar()
cbar.set_label('Insolation [Earth Flux]')
plt.show()

Answer 1

Вы можете выбрать обрезанную часть вашего фрейма данных следующим образом:

data2_trimmed=data2.loc[data2['pl_insol']<=1.5,:]
z = data2_trimmed['pl_insol']
x = data2_trimmed['pl_rade']
y = data2_trimmed['pl_orbper']

, где вы выбираете все столбцы и только те строки, которые удовлетворяют условию data2['pl_insol']<=1.5