Matplotlib scatterplot; color เป็นฟังก์ชันของตัวแปรตัวที่สาม

ฉันต้องการสร้าง scatterplot (โดยใช้ matplotlib) ซึ่งเป็นจุดที่แรเงาตามตัวแปรตัวที่สาม ฉันสนิทกับสิ่งนี้มาก:

plt.scatter(w, M, c=p, marker='s')

โดยที่ w และ M เป็นดาต้าพอยน์และ p เป็นตัวแปรที่ฉันต้องการแรเงาด้วยความเคารพ
อย่างไรก็ตามฉันต้องการที่จะทำมันในเฉดสีเทามากกว่าสี ใครช่วยได้บ้าง

ไม่จำเป็นต้องตั้งค่าสีด้วยตนเอง ให้ระบุ colormap สีเทาแทน ...

import numpy as np
import matplotlib.pyplot as plt

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

# Plot...
plt.scatter(x, y, c=y, s=500)
plt.gray()

plt.show()

หรือหากคุณต้องการช่วงกว้างของ colormapsคุณยังสามารถระบุcmapkwarg scatterไป หากต้องการใช้เวอร์ชันที่กลับด้านของสิ่งเหล่านี้เพียงระบุ_rเวอร์ชัน "" ของสิ่งเหล่านี้ เช่นแทนgray_r grayมีหลาย colormaps สีเทาที่แตกต่างกันที่ทำไว้ล่วงหน้าเป็น (เช่นgray, gist_yarg, binaryฯลฯ )

import matplotlib.pyplot as plt
import numpy as np

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

plt.scatter(x, y, c=y, s=500, cmap='gray')
plt.show()

ในสีเทา matplotlib สามารถกำหนดเป็นสตริงของค่าตัวเลขระหว่าง 0-1
ตัวอย่างเช่นc = '0.1'

จากนั้นคุณสามารถแปลงตัวแปรตัวที่สามของคุณเป็นค่าภายในช่วงนี้และใช้มันเพื่อแต้มสีของคุณ
ในตัวอย่างต่อไปนี้ฉันใช้ตำแหน่ง y ของจุดเป็นค่าที่กำหนดสี:

from matplotlib import pyplot as plt

x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]

color = [str(item/255.) for item in y]

plt.scatter(x, y, s=500, c=color)

plt.show()

บางครั้งคุณอาจจะต้องมีสีที่พล็อตได้อย่างแม่นยำขึ้นอยู่กับกรณีที่ x-ค่า ตัวอย่างเช่นคุณอาจมี dataframe ที่มีตัวแปร 3 ประเภทและบางจุดข้อมูล และคุณต้องการที่จะทำตาม

• พล็อตจุดที่สอดคล้องกับตัวแปรทางกายภาพ 'A' ในสีแดง
• พล็อตจุดที่สอดคล้องกับตัวแปรทางกายภาพ 'B' ใน BLUE
• พล็อตจุดที่สอดคล้องกับตัวแปรทางกายภาพ 'C' ใน GREEN

ในกรณีนี้คุณอาจต้องเขียนลงฟังก์ชั่นแบบสั้นเพื่อแมปค่า x กับชื่อสีที่เกี่ยวข้องเป็นรายการแล้วส่งผ่านรายการนั้นไปยังplt.scatterคำสั่ง

x=['A','B','B','C','A','B']
y=[15,30,25,18,22,13]

# Function to map the colors as a list from the input list of x variables
def pltcolor(lst):
    cols=[]
    for l in lst:
        if l=='A':
            cols.append('red')
        elif l=='B':
            cols.append('blue')
        else:
            cols.append('green')
    return cols
# Create the colors list using the function above
cols=pltcolor(x)

plt.scatter(x=x,y=y,s=500,c=cols) #Pass on the list created by the function here
plt.grid(True)
plt.show()