Bagaimana untuk Mencipta Plot Scatter dengan Data Kategori dalam Matplotlib Python?

Cara Membuat Plot Serakan mengikut Kategori

Dalam Matplotlib Python, mencipta plot serakan mengikut kategori boleh dicapai menggunakan kaedah plot, seperti yang ditunjukkan di bawah:

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt

# Generate Data
num = 20
x, y = np.random.random((2, num))
labels = np.random.choice(['a', 'b', 'c'], num)
df = pd.DataFrame(dict(x=x, y=y, label=labels))

# Group Data
groups = df.groupby('label')

# Plot
fig, ax = plt.subplots()
ax.margins(0.05)  # Optional padding
for name, group in groups:
    ax.plot(group.x, group.y, marker='o', linestyle='', ms=12, label=name)
ax.legend()

plt.show()

Untuk penampilan yang lebih tersuai menyerupai lalai Pandas gaya:

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd

# Generate Data
num = 20
x, y = np.random.random((2, num))
labels = np.random.choice(['a', 'b', 'c'], num)
df = pd.DataFrame(dict(x=x, y=y, label=labels))

# Group Data
groups = df.groupby('label')

# Plot
plt.rcParams.update(pd.tools.plotting.mpl_stylesheet)
colors = pd.tools.plotting._get_standard_colors(len(groups), color_type='random')

fig, ax = plt.subplots()
ax.set_color_cycle(colors)
ax.margins(0.05)
for name, group in groups:
    ax.plot(group.x, group.y, marker='o', linestyle='', ms=12, label=name)
ax.legend(numpoints=1, loc='upper left')

plt.show()

Atas ialah kandungan terperinci Bagaimana untuk Mencipta Plot Scatter dengan Data Kategori dalam Matplotlib Python?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!