从 python 中的 csv 数据集中提取 ID 和相关数据

Question

Making a program for my Final Year Project. Program takes the longitude and latitude coords from a.csv dataset and plots them on the map. Issue I am having is there is multiple ID's and this totals 445,000+ points.

How would I refine it so the program can differentiate between the IDs?

 def create_image(self, color, width=2):
    # Creates an image that contains the Map and the GPS record
    # color = color the GPS line is
    # width = width of the GPS line
    data = pd.read_csv(self.data_path, header=0)
    # sep will separate the latitude from the longitude
    data.info()
    self.result_image = Image.open(self.map_path, 'r')
    img_points = []
    gps_data = tuple(zip(data['latitude'].values, data['longitude'].values))

    for d in gps_data:
        x1, y1 = self.scale_to_img(d, (self.result_image.size[0], self.result_image.size[1]))
        img_points.append((x1, y1))
    draw = ImageDraw.Draw(self.result_image)
    draw.line(img_points, fill=color, width=width)

I have also attached the github project here the program works but I am just trying to minimize how many users it plots at once.

Thanks in advance.

Answer 1

To check for a specific ID you could create a filter. For this dataframe

    long    lat     ID
0   10      5       test1
1   15      20      test2

you could do the following:

id_filt = df_data['ID'] == 'test1'

This can be used to filter out every entry from the dataframe that has the ID 'test1'

df_data[id_filt]


long    lat     ID
10      5       test1