def add()

in viz/utils.py [0:0]

• 22 lines of code
• 12 McCabe index (conditional complexity)

    def add(self, xs, ys, std, label):
        valid = False
        y_data = {}
        std_data = {}
        xs = [round(x, 2) for x in xs]
        for i, x in enumerate(xs):
            if x in self.xarr:
                valid = True
                y_data[x] = ys[i]
                std_data[x] = std[i]

        if valid:
            self.mean[label] = np.mean(np.array([v for k, v in y_data.items()]))
            if self.include_std:
                self.std[label] = np.std(
                    np.array([v for k, v in y_data.items()])
                )
            for x in self.xarr:
                if x not in y_data:
                    y_data[x] = "-"
                    std_data[x] = "-"

            self.ys[label] = [y_data[x] for x in self.xarr]
            self.stds[label] = [std_data[x] for x in self.xarr]