Improve plot map func

[ShawnChen09]

No description provided.

[ShawnChen09]

Question:
Is the type of input_data a pd.DataFrame? I tried to access a column's values using indexing in the format input_data[1] or input_data[1:3], but it doesn't work. Therefore, I changed the indexing format to input_data.iloc[:, i], where i represents the column index.

[ShawnChen09]

Update Details:
I have added two new arguments. The first one is z_type (I am considering renaming it to z_dim). The allowed options are '1d', '2d', and '3d'. The '2d' option will be transformed into '3d' by filling the values of the third axis with 0. The functionality for '1d' remains unchanged. For both '2d' and '3d', the values will be normalized within the range (0, 1) across the three dimensions, representing the (R, G, B) values of the point's color.

The second argument is cax, which accepts a list of floats to control the colorbar's position (left, bottom, width, height). If set to None, the default settings will be used. For z_type = '1d', a colorbar still be used to display the linear relationship, while for '2d' and '3d', a scatter plot will be used to show the distribution of values in RGB space.