Method plot

OGP_v11.py:2926–2965

(self, strike_column, dip_column, color_column=None)

        self.hole_ids = hole_ids

    def plot(self, strike_column, dip_column, color_column=None):
        # Filter DataFrame based on selected hole IDs
        df = self.df[self.df['hole_id'].isin(self.hole_ids)]

        # Convert strike angles to radians
        strike_angles = np.radians(df[strike_column])

        # Determine colors for each row in DataFrame
        if color_column:
            unique_values = df[color_column].unique()
            colormap = plt.cm.get_cmap('viridis', len(unique_values))
            color_dict = dict(zip(unique_values, [colormap(i) for i in range(len(unique_values))]))
            colors = df[color_column].map(color_dict)
        else:
            colors = 'blue'

        # Create rose diagram
        plt.figure(figsize=(8, 8))
        ax = plt.subplot(111, projection='polar')

        # Bin strike angles and get the counts for each bin
        strike_bins = np.linspace(0, 2*np.pi, 36)
        counts, bin_edges = np.histogram(strike_angles, bins=strike_bins)
        
        # Compute bin centers
        bin_centers = (bin_edges[:-1] + bin_edges[1:]) / 2.0

        ax.bar(bin_centers, counts, color=colors, width=2*np.pi/36)

        ax.set_theta_zero_location('N')
        ax.set_theta_direction(-1)

        # If there is color coding, add a legend
        if color_column:
            legend_handles = []
            for category, color in color_dict.items():
                legend_handles.append(mpatches.Patch(color=color, label=category))
            ax.legend(handles=legend_handles, loc='upper left')

        plt.show()


    def dialog(self):