Example #1
0
def _create_chart(worksheet):
    """Create the f*****g chart"""
    chart = ScatterChart()
    chart.varyColors = True
    chart.title = "Financial Analysis"
    chart.style = 1
    chart.height = 10
    chart.width = 20
    chart.x_axis.title = "Financial Quarter"
    chart.y_axis.title = "Cost"
    chart.legend = None
    chart.x_axis.majorUnit = 0.5
    chart.x_axis.minorGridlines = None
    #   chart.y_axis.majorUnit = 200

    xvalues = Reference(worksheet, min_col=1, min_row=3, max_row=6)
    picker = _color_gen()
    for i in range(2, 7):
        values = Reference(worksheet, min_col=i, min_row=2, max_row=6)
        series = Series(values, xvalues, title_from_data=True)
        series.smooth = True
        series.marker.symbol = "circle"
        line_prop = LineProperties(solidFill=next(picker))
        series.graphicalProperties.line = line_prop
        chart.series.append(series)
    worksheet.add_chart(chart, "G1")
    return worksheet
Example #2
0
def draw_chart(wb_path: str):
    wb = openpyxl.load_workbook(wb_path)
    ws = wb.active
    c1 = ScatterChart()
    c1.y_axis.majorGridlines = None
    c1.x_axis.majorGridlines = None
    c1.x_axis.tickLblSkip = 100
    #data = Reference(ws, min_col=1, min_row=1, max_col=ws.max_column, max_row=ws.max_row)
    #c1.add_data(data, titles_from_data=True)

    skips = ws.max_column / 2
    col = 1
    for i in range(1, ws.max_column, 2):
        xvalues = Reference(ws, min_col=i, min_row=1, max_row=ws.max_row)
        values = Reference(ws, min_col=i + 1, min_row=1, max_row=ws.max_row)
        series = Series(values, xvalues, title_from_data=True)
        series.smooth = True
        c1.series.append(series)

    ws.add_chart(c1, "D10")
    wb.save(wb_path)
    return True
Example #3
0
def draw_ordered_ranks(worksheet):
    chart = LineChart()
    chart.title = 'Ranks in descending order'
    chart.x_axis.title = 'Function'
    chart.y_axis.title = 'Rank'
    chart.style = 11

    i = 0
    while True:
        a = alpha(worksheet, i)
        if a is None:
            break
        col = 3 * i + 2
        data = Reference(worksheet,
                         min_col=col,
                         max_col=col,
                         min_row=2,
                         max_row=min(worksheet.max_row, 100))
        series = Series(data, title='a=' + a)
        series.smooth = True
        series.graphicalProperties.line.width = 20000
        chart.append(series)
        i += 1
    worksheet.add_chart(chart, 'D25')
            #line.x_axis.scaling.max = int(max(origYPSNR_dict[filename][0] , testYPSNR_dict[filename][0],
            #                                  testYPSNR_dict2[filename][0], testYPSNR_dict3[filename][0])) + 2  # y轴的最大值

        oriXdata = Reference(
            sheet_anchor,
            min_col=4,
            min_row=excelCurrRow + (index_num - 1) * dataVerStep,
            max_row=excelCurrRow + (index_num - 1) * dataVerStep + 3)
        oriYdata = Reference(
            sheet_anchor,
            min_col=5,
            min_row=excelCurrRow + (index_num - 1) * dataVerStep,
            max_row=excelCurrRow + (index_num - 1) * dataVerStep + 3)
        series = Series(oriYdata, oriXdata, title=anchor_codec)
        series.marker.symbol = 'circle'
        series.smooth = True
        line.series.append(series)

        testXdata = Reference(
            sheet_refer,
            min_col=4,
            min_row=excelCurrRow + (index_num - 1) * dataVerStep,
            max_row=excelCurrRow + (index_num - 1) * dataVerStep + 3)
        testYdata = Reference(
            sheet_refer,
            min_col=5,
            min_row=excelCurrRow + (index_num - 1) * dataVerStep,
            max_row=excelCurrRow + (index_num - 1) * dataVerStep + 3)
        series = Series(testYdata, testXdata, title=refer1_codec)
        series.marker.symbol = 'circle'
        series.smooth = True
def graph(string: str):

    jsonOBJ = json.loads(string[string.find(BeginJSON) + len(BeginJSON) : string.find(EndJSON)])
    dataStream = string[string.find(EndJSON) + len(EndJSON) :]
    StrDict = {v: k for k, v in jsonOBJ[1].items()}

    data = dict()

    dataTitles = list(["Elapsed Time (s)"])

    epochSmol = -1

    for line in dataStream.splitlines():
        try:
            msg = line.split(" ")
            if len(msg) != 4 or not StrDict.get(int(msg[2])):
                continue
            msgID = StrDict[int(msg[2])]
            if not data.get(msgID):
                data[msgID] = list()
                dataTitles.append(msgID)
            epoch = int(msg[0])
            data[msgID].append(list((epoch, int(msg[3]), data[msgID])))

            if epochSmol == -1 or epoch < epochSmol:
                epochSmol = epoch

        except KeyError as e:
            print("KeyError: ", e)
            pass

    for _, value in data.items():
        value.sort(key=lambda x: x[0])
        for lst in value:
            lst[0] -= epochSmol
            lst[0] /= 1000000000

    wb = openpyxl.Workbook()
    ws = wb.active

    ws.append(dataTitles)

    rows = list()

    def getRow():  # IMPROVE: better series generation
        row = list()
        for title in dataTitles:
            column = data.get(title)
            if not column or len(column) == 0:
                row.append((epochSmol + 1, 0))
            else:
                row.append(column[0])

        smallestEpoch = epochSmol

        for item in row:
            if item[0] < smallestEpoch:
                smallestEpoch = item[0]

        if smallestEpoch == epochSmol:
            return

        finalRow = list([smallestEpoch])
        row.pop(0)

        for item in row:
            if item[0] != smallestEpoch:
                finalRow.append(item[1])
            else:
                finalRow.append(item[2].pop(0)[1])

        if len(finalRow) == 1:
            return

        return finalRow

    while True:
        row = getRow()
        if row:
            rows.append(row)
        else:
            break

    for row in rows:
        ws.append(row)

    chart = ScatterChart()
    chart.title = "Interpreted Data"
    chart.style = 2
    chart.x_axis.title = "Elapsed Time (s)"
    chart.y_axis.title = "Value"
    chart.height = 20
    chart.width = 45

    xvalues = Reference(ws, min_col=1, min_row=2, max_row=len(rows) + 1)

    for i in range(2, len(dataTitles) + 1):
        values = Reference(ws, min_col=i, min_row=1, max_row=len(rows) + 1)
        series = Series(values, xvalues, title_from_data=True)
        # series.marker.symbol = "circle"
        # series.marker.size = 3
        # series.graphicalProperties.line.noFill = True
        series.smooth = True
        series.graphicalProperties.line.width = 10000  # width in EMUs
        chart.series.append(series)

    ws.add_chart(chart, "A1")
    wb.save("{}.xlsx".format(SaveName))
Example #6
0
def plot(ws_Excel_Unite_1_exp,
         ws_Excel_Unite_2_exp,
         ws_Excel_Unite_exp_path,
         dir_file_Excel_Unite,
         NameGenInvestigated):
    global n, value_to_time, name_obj_gen, h

    ws_Excel_Unite_exp_path.cell(
        column=1,
        row=1,
        value=f"Загруженный файл {dir_file_Excel_Unite}"
    )
    num_time_to_row = 0
    col_count = ws_Excel_Unite_1_exp.max_column - 1
    col_gen = 0

    for n in range(1, 50):
        value_to_time = ws_Excel_Unite_1_exp[f"A{n}"].value
        if value_to_time == ' TIME':
            num_time_to_row = int(n) + 1

    for h in range(2, col_count + 1):
        name_obj_gen = ws_Excel_Unite_1_exp[f"{get_column_letter(h)}{str(num_time_to_row - 1)}"].value
        if name_obj_gen == f"P/{NameGenInvestigated}":
            col_gen = h
            break

    for g in range(2, col_count + 1):
        nameObjExp1 = ws_Excel_Unite_1_exp[f"{get_column_letter(g)}{str(num_time_to_row - 1)}"].value
        nameObjExp2 = ws_Excel_Unite_2_exp[f"{get_column_letter(g)}{str(num_time_to_row - 1)}"].value
        nameObjExp1_dict = dict_set[str(nameObjExp1)]
        nameObjExp2_dict = dict_set[str(nameObjExp2)]
        ws_Excel_Unite_1_exp.cell(column=g,
                                  row=num_time_to_row - 1,
                                  value=f"с PSS {nameObjExp1_dict}")

        ws_Excel_Unite_2_exp.cell(column=g,
                                  row=num_time_to_row - 1,
                                  value=f"без PSS {nameObjExp2_dict}")

    max_Chart = max_var_object(ws_Excel_Unite_1_exp, col_gen, num_time_to_row)
    min_Chart = min_var_object(ws_Excel_Unite_1_exp, col_gen, num_time_to_row)

    maxCh60 = max_Ch60(ws_Excel_Unite_1_exp, col_gen, num_time_to_row)
    minCh60 = min_Ch60(ws_Excel_Unite_1_exp, col_gen, num_time_to_row)

    # График 1 активной мощности СГ на интервале от 0 до 15
    ch4 = ScatterChart()

    xvalues_exp1 = Reference(ws_Excel_Unite_1_exp,
                             min_col=1,
                             min_row=num_time_to_row,
                             max_row=ws_Excel_Unite_1_exp.max_row)

    values_exp1 = Reference(ws_Excel_Unite_1_exp,
                            min_col=int(col_gen),
                            min_row=num_time_to_row - 1,
                            max_row=ws_Excel_Unite_1_exp.max_row)

    series_exp1 = Series(values_exp1, xvalues_exp1, title_from_data=True)
    name_object = ws_Excel_Unite_1_exp[f'{get_column_letter(col_gen)}{str(num_time_to_row - 1)}'].value
    remove_list = ['с', 'PSS']
    edit_str_as_list = name_object.split()
    final_list = [word for word in edit_str_as_list if word not in remove_list]
    final_str = ' '.join(final_list)
    final_list.clear()
    ch4.title = f"{final_str}"
    ch4.x_axis.title = "Время, с"  # название оси Х
    ch4.y_axis.title = "Активная мощность, МВт"  # название оси У
    # ch4.style = 12 # стиль диаграммы
    ch4.series.append(series_exp1)

    ch4.x_axis.scaling.min = 49
    ch4.y_axis.scaling.min = 0
    ch4.y_axis.scaling.max = max_Chart + 10
    ch4.x_axis.scaling.max = 65

    ws_Excel_Unite_exp_path.add_chart(ch4, f'{get_column_letter(1)}{str(5)}')

    # График 2 активной мощности СГ на интервале от 0 до 15
    ch2 = ScatterChart()

    xvalues_exp1 = Reference(ws_Excel_Unite_1_exp,
                             min_col=1,
                             min_row=num_time_to_row,
                             max_row=ws_Excel_Unite_1_exp.max_row)

    values_exp1 = Reference(ws_Excel_Unite_1_exp,
                            min_col=int(col_gen),
                            min_row=num_time_to_row - 1,
                            max_row=ws_Excel_Unite_1_exp.max_row)

    xvalues_exp2 = Reference(ws_Excel_Unite_2_exp,
                             min_col=1,
                             min_row=num_time_to_row,
                             max_row=ws_Excel_Unite_2_exp.max_row)

    values_exp2 = Reference(ws_Excel_Unite_2_exp,
                            min_col=int(col_gen),
                            min_row=num_time_to_row - 1,
                            max_row=ws_Excel_Unite_2_exp.max_row)

    series_exp1 = Series(values_exp1, xvalues_exp1, title_from_data=True)
    series_exp2 = Series(values_exp2, xvalues_exp2, title_from_data=True)

    name_object = ws_Excel_Unite_1_exp[f'{get_column_letter(col_gen)}{str(num_time_to_row - 1)}'].value
    remove_list = ['с', 'PSS']
    edit_str_as_list = name_object.split()
    final_list1 = [word for word in edit_str_as_list if word not in remove_list]
    final_str = ' '.join(final_list1)
    final_list1.clear()
    # ch2.style = 12 # стиль диаграммы
    ch2.title = f"{final_str}"
    ch2.x_axis.title = "Время, с"  # название оси Х
    ch2.y_axis.title = "Активная мощность, МВт"  # название оси У

    series_exp2.smooth = False  # сграживание кривой
    series_exp2.graphicalProperties.line.width = 350  # толщина кривой линии
    series_exp2.graphicalProperties.line.solidFill = 'FF0000'  # цвет кривой Red

    ch2.series.append(series_exp1)
    ch2.series.append(series_exp2)

    ch2.x_axis.scaling.min = 49
    ch2.y_axis.scaling.min = 0
    ch2.x_axis.scaling.max = 65

    ws_Excel_Unite_exp_path.add_chart(ch2, get_column_letter(11) + str(22))

    # График 3 активной мощности СГ на интервале от 1 до 6
    ch_PSS = ScatterChart()

    xvalues_exp1_pss = Reference(ws_Excel_Unite_1_exp,
                                 min_col=1,
                                 min_row=num_time_to_row,
                                 max_row=ws_Excel_Unite_1_exp.max_row)

    values_exp1_pss = Reference(ws_Excel_Unite_1_exp,
                                min_col=int(col_gen),
                                min_row=num_time_to_row - 1,
                                max_row=ws_Excel_Unite_1_exp.max_row)

    xvalues_exp2_pss = Reference(ws_Excel_Unite_2_exp,
                                 min_col=1,
                                 min_row=num_time_to_row,
                                 max_row=ws_Excel_Unite_2_exp.max_row)

    values_exp2_pss = Reference(ws_Excel_Unite_2_exp,
                                min_col=int(col_gen),
                                min_row=num_time_to_row - 1,
                                max_row=ws_Excel_Unite_2_exp.max_row)

    series_exp1_pss = Series(values_exp1_pss,
                             xvalues_exp1_pss,
                             title_from_data=True)

    series_exp2_pss = Series(values_exp2_pss,
                             xvalues_exp2_pss,
                             title_from_data=True)

    name_object = ws_Excel_Unite_1_exp[f'{get_column_letter(col_gen)}{str(num_time_to_row - 1)}'].value

    remove_list = ['с', 'PSS']
    edit_str_as_list = name_object.split()
    final_list2 = [word for word in edit_str_as_list if word not in remove_list]
    final_str = ' '.join(final_list2)
    final_list2.clear()
    ch_PSS.title = f"Эффективность PSS(стаб.) {final_str}"
    ch_PSS.x_axis.title = "Время, с"  # название оси Х
    ch_PSS.y_axis.title = "Активная мощность, МВт"  # название оси У

    series_exp2_pss.smooth = False  # сграживание кривой
    series_exp2_pss.graphicalProperties.line.width = 350  # сграживание кривой
    series_exp2_pss.graphicalProperties.line.solidFill = 'FF0000'  # цвет кривой Red

    ch_PSS.series.append(series_exp1_pss)
    ch_PSS.series.append(series_exp2_pss)

    ch_PSS.x_axis.scaling.min = 51
    ch_PSS.y_axis.scaling.min = int(min_Chart)
    ch_PSS.y_axis.scaling.max = int(max_Chart + 4)
    ch_PSS.x_axis.scaling.max = 56

    ws_Excel_Unite_exp_path.add_chart(ch_PSS, f'{get_column_letter(21)}{str(22)}')

    # График 4 активной мощности СГ на интервале от 11 до 26
    ch3 = ScatterChart()
    xvalues = Reference(ws_Excel_Unite_1_exp,
                        min_col=1,
                        min_row=num_time_to_row,
                        max_row=ws_Excel_Unite_1_exp.max_row)

    values = Reference(ws_Excel_Unite_1_exp,
                       min_col=col_gen,
                       min_row=num_time_to_row - 1,
                       max_row=ws_Excel_Unite_1_exp.max_row)

    series = Series(values, xvalues, title_from_data=True)

    name_object = ws_Excel_Unite_1_exp[f'{get_column_letter(col_gen)}{str(num_time_to_row - 1)}'].value

    remove_list = ['с', 'PSS']
    edit_str_as_list = name_object.split()
    final_list3 = [word for word in edit_str_as_list if word not in remove_list]
    final_str = ' '.join(final_list3)
    final_list3.clear()
    # ch3.style = 12    # стиль диаграммы
    ch3.title = f"{final_str}"
    ch3.x_axis.title = "Время, с"  # название оси Х
    ch3.y_axis.title = "Активная мощность, МВт"  # название оси У
    ch3.series.append(series)

    ch3.x_axis.scaling.min = 61
    ch3.y_axis.scaling.min = int(minCh60)
    ch3.y_axis.scaling.max = int(maxCh60 + 1)
    ch3.x_axis.scaling.max = 76

    ws_Excel_Unite_exp_path.add_chart(ch3, get_column_letter(1) + str(22))

    for k in range(2, col_count + 1):
        ch1 = ScatterChart()
        xvalues_exp1 = Reference(ws_Excel_Unite_1_exp,
                                 min_col=1,
                                 min_row=num_time_to_row,
                                 max_row=ws_Excel_Unite_1_exp.max_row)

        values_exp1 = Reference(ws_Excel_Unite_1_exp,
                                min_col=k,
                                min_row=num_time_to_row - 1,
                                max_row=ws_Excel_Unite_1_exp.max_row)

        xvalues_exp2 = Reference(ws_Excel_Unite_2_exp,
                                 min_col=1,
                                 min_row=num_time_to_row,
                                 max_row=ws_Excel_Unite_2_exp.max_row)

        values_exp2 = Reference(ws_Excel_Unite_2_exp,
                                min_col=k,
                                min_row=num_time_to_row - 1,
                                max_row=ws_Excel_Unite_2_exp.max_row)

        series_exp1 = Series(values_exp1, xvalues_exp1, title_from_data=True)
        series_exp2 = Series(values_exp2, xvalues_exp2, title_from_data=True)

        name_object = ws_Excel_Unite_1_exp[get_column_letter(k) + str(num_time_to_row - 1)].value
        remove_list = ['с', 'PSS']
        edit_str_as_list = name_object.split()
        final_list4 = [word for word in edit_str_as_list if word not in remove_list]
        final_str = ' '.join(final_list4)
        final_list4.clear()
        # ch1.style = 12    # стиль диаграммы
        ch1.title = f"{final_str}"
        ch1.x_axis.title = "Время, с"  # название оси Х
        ch1.y_axis.title = "Активная мощность, МВт"  # название оси У

        series_exp2.smooth = False  # сграживание кривой
        series_exp2.graphicalProperties.line.width = 350  # сграживание кривой
        series_exp2.graphicalProperties.line.solidFill = 'FF0000'  # цвет кривой Red

        series_exp2.smooth = False  # сграживание кривой
        series_exp2.graphicalProperties.line.width = 350  # сграживание кривой
        series_exp2.graphicalProperties.line.solidFill = 'FF0000'  # цвет кривой Red

        ch1.series.append(series_exp1)
        ch1.series.append(series_exp2)

        ch1.x_axis.scaling.min = 48
        ch1.y_axis.scaling.min = 0
        ch1.x_axis.scaling.max = 80
        # ch1.y_axis.scaling.max = 650 # диапазоны на графике

        size_plot = 10
        if k < int(col_count / 2):
            if k == 2:
                ws_Excel_Unite_exp_path.add_chart(ch1, get_column_letter(1) + str(40))
            elif k != 2:
                ws_Excel_Unite_exp_path.add_chart(ch1, get_column_letter((k - 2) * size_plot) + str(40))
        else:
            ws_Excel_Unite_exp_path.add_chart(ch1, get_column_letter((k - 2) * size_plot) + str(40))