如何使用来自两个词典的数据创建散点图？

小能豆

如何使用来自两个词典的数据创建散点图？

在我的代码中，用户导入一个包含四列且行数可变的数据文件。第一列包含动物的名称，第二列包含其在田地中的 x 位置，第三列包含其 y 位置，第四列包含其 z 位置。

#load the data
emplaced_animals_data = np.genfromtxt('animal_data.txt', skip_header = 1, dtype = str)
print(type(emplaced_animals_data))
print(emplaced_animals_data)

<class 'numpy.ndarray'>
[['butterfly' '1' '1' '3']
 ['butterfly' '2' '2' '3']
 ['butterfly' '3' '3' '3']
 ['dragonfly' '4' '1' '1']
 ['dragonfly' '5' '2' '1']
 ['dragonfly' '6' '3' '1']
 ['cat' '4' '4' '2']
 ['cat' '5' '5' '2']
 ['cat' '6' '6' '2']
 ['cat' '7' '8' '3']
 ['elephant' '8' '9' '3']
 ['elephant' '9' '10' '4']
 ['elephant' '10' '10' '4']
 ['camel' '10' '11' '5']
 ['camel' '11' '6' '5']
 ['camel' '12' '5' '6']
 ['camel' '12' '3' '6']
 ['bear' '13' '13' '7']
 ['bear' '5' '15' '7']
 ['bear' '4' '10' '5']
 ['bear' '6' '9' '2']
 ['bear' '15' '13' '1']
 ['dog' '1' '3' '9']
 ['dog' '2' '12' '8']
 ['dog' '3' '10' '1']
 ['dog' '4' '8' '1']]

我使用字典创建了一组键和值。我的键是动物，值是它们的位置。我为 X、Y 和 Z 位置创建了一个字典。

animal_list = ['cat', 'elephant', 'camel', 'bear', 'dog']

locsX = []
locsY = []
locsZ = []
animalsX = {}
animalsY = {}
animalsZ = {}

for i in range(0, len(animal_list)):
    for j in range(0, len(emplaced_animals_data)):
        for k in range(0, len(animal_list)):
            if animal_list[i] == animal_list[k] and animal_list[i] == emplaced_animals_data[j,0]:
                locsX = np.append(locsX, emplaced_animals_data[j,1])
                locsY = np.append(locsY, emplaced_animals_data[j,2])
                locsZ = np.append(locsZ, emplaced_animals_data[j,3])
                animalsX.update({animal_list[k]:locsX})
                animalsY.update({animal_list[k]:locsY})
                animalsZ.update({animal_list[k]:locsZ})
print(animalsX)
print(animalsY)


{'cat': array(['4', '5', '6', '7'], dtype='<U32'), 'elephant': array(['4', '5', '6', '7', '8', '9', '10'], dtype='<U32'), 'camel': array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12'],
      dtype='<U32'), 'bear': array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12', '13',
       '5', '4', '6', '15'], dtype='<U32'), 'dog': array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12', '13',
       '5', '4', '6', '15', '1', '2', '3', '4'], dtype='<U32')}
{'cat': array(['4', '5', '6', '8'], dtype='<U32'), 'elephant': array(['4', '5', '6', '8', '9', '10', '10'], dtype='<U32'), 'camel': array(['4', '5', '6', '8', '9', '10', '10', '11', '6', '5', '3'],
      dtype='<U32'), 'bear': array(['4', '5', '6', '8', '9', '10', '10', '11', '6', '5', '3', '13',
       '15', '10', '9', '13'], dtype='<U32'), 'dog': array(['4', '5', '6', '8', '9', '10', '10', '11', '6', '5', '3', '13',
       '15', '10', '9', '13', '3', '12', '10', '8'], dtype='<U32')}

如何使用字典中每个键（动物）的 X 和 Y 位置值来创建散点图？我希望每个键（动物）的数据点具有不同的颜色。

阅读 14

2024-12-11

共1个答案

小能豆

要使用每只动物的 X 和 Y 位置值创建散点图并为每只动物分配不同的颜色，您可以使用matplotlib来绘制数据。您创建的animalsX和animalsY字典包含每只动物的 X 和 Y 坐标，您可以将它们绘制在散点图上。

下面是一个示例，说明如何修改代码以创建散点图，其中每种动物都以不同的颜色显示：

确保数据格式正确（为了正确绘图，从字符串转换为浮点数）。
用于matplotlib绘制数据，为每只动物分配不同的颜色。

具体操作如下：

import numpy as np
import matplotlib.pyplot as plt

# Load the data (assuming 'animal_data.txt' is already loaded in emplaced_animals_data)
# emplaced_animals_data = np.genfromtxt('animal_data.txt', skip_header=1, dtype=str)

# Example of animals' location dictionaries
animalsX = {'cat': np.array(['4', '5', '6', '7']),
             'elephant': np.array(['4', '5', '6', '7', '8', '9', '10']),
             'camel': np.array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12']),
             'bear': np.array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12', '13']),
             'dog': np.array(['4', '5', '6', '7', '8', '9', '10', '10', '11', '12', '12', '13', '5', '4'])}

animalsY = {'cat': np.array(['4', '5', '6', '8']),
             'elephant': np.array(['4', '5', '6', '8', '9', '10', '10']),
             'camel': np.array(['4', '5', '6', '8', '9', '10', '10', '11', '6']),
             'bear': np.array(['4', '5', '6', '8', '9', '10', '10', '11', '6']),
             'dog': np.array(['4', '5', '6', '8', '9', '10', '10', '11', '6', '5'])}

# Assign colors to each animal
colors = {'cat': 'blue', 'elephant': 'green', 'camel': 'red', 'bear': 'purple', 'dog': 'orange'}

# Create the plot
plt.figure(figsize=(8, 8))

# Loop through each animal and plot its locations
for animal in animalsX:
    x_coords = animalsX[animal].astype(float)  # Convert strings to floats
    y_coords = animalsY[animal].astype(float)  # Convert strings to floats
    plt.scatter(x_coords, y_coords, color=colors[animal], label=animal)

# Add labels and title
plt.xlabel('X Location')
plt.ylabel('Y Location')
plt.title('Animal Locations')

# Add a legend
plt.legend()

# Show the plot
plt.show()

主要变化：

将字符串转换为浮点数：由于您的数据是字符串格式 ( dtype='<U32')，因此您需要将它们转换为浮点数以进行绘图。这通过astype(float)X 和 Y 坐标来完成。
使用不同颜色进行绘图：每种动物都有一种独特的颜色，在colors字典中定义。该plt.scatter()函数用于绘制数据，其中每种动物的 X 和 Y 坐标都用其相应的颜色绘制。
图例：该plt.legend()函数为图表添加图例，以便识别每种动物的颜色。

这将生成一个散点图，根据 X 和 Y 位置为每只动物绘制不同的颜色。

2024-12-11