From bd8f94353a03fc042201b6543841c9174fe0f857 Mon Sep 17 00:00:00 2001 From: corizhang Date: Tue, 7 Nov 2023 17:28:34 +0800 Subject: [PATCH] new --- test.py | 3 ++ 创建wav.py | 99 +++++++++++++++++++++++++++++++++--------------------- 2 files changed, 64 insertions(+), 38 deletions(-) create mode 100644 test.py diff --git a/test.py b/test.py new file mode 100644 index 0000000..5309360 --- /dev/null +++ b/test.py @@ -0,0 +1,3 @@ +tlist = [] +tlist.insert(0,'abc') +print(tlist) \ No newline at end of file diff --git a/创建wav.py b/创建wav.py index 28113b5..580ce64 100644 --- a/创建wav.py +++ b/创建wav.py @@ -3,48 +3,66 @@ import pandas as pd import wave import struct + +# 补全章节位数 +def add_zero_to_title(chapterName, places): + """ + :param chapterName: 章节名称 + :param places: 章节编号规定的位数, 如003, 即命名时需要的位数为3位 + :return: 补全位数的章节编号 + """ + while len(chapterName) < places: + chapterName = "0" + chapterName + return chapterName + + # 打开Excel文件 -def add_zero_to_title(title): - while len(title) < 3: - title = "0" + title - return title - -def get_data(file_path, column_list, cv_name): - excel_data = pd.read_excel(file_path) - column_names = excel_data.columns.tolist() - title = column_names[0] - excel_data[title] = excel_data[title].ffill() - column_cv = column_names[2] - print(column_cv) - filter_value = cv_name - filtered_data = excel_data[excel_data[column_cv].astype(str).str.contains(filter_value, na=False)] - print(filtered_data) - +def get_wav_title(filePath, selectedColumnList, sheetName, cvName, places): + """ + :param filePath: 剧组文件的路径 + :param selectedColumnList: 选择要保留的数据列 + :param sheetName: 要读取的sheet名称 + :param cvName: cv名称 + :return: + """ + role_data = pd.read_excel(filePath, sheet_name=sheetName) + role_data[selectedColumnList[0]] = role_data[selectedColumnList[0]].ffill() + filtered_role_data = role_data[role_data[selectedColumnList[1]].astype(str).str.contains(cvName, na=False)] + # filtered_role_data = role_data[role_data[selectedColumnList[1]].str.contains(cvName, na=False)] + print(filtered_role_data) wav_title_list = [] + # 逐行读取指定列的数据 - for index, row in filtered_data.iterrows(): - title = row[column_list[0]] + for index, row in filtered_role_data.iterrows(): + title = row[selectedColumnList[0]] title = re.findall(r"\d+\.?\d*", title) - title = add_zero_to_title(title[0]) - role = row[column_list[1]] - num = row[column_list[2]] - print(num) - # wav_title = title[0] + '-' + role + '-' + cv_name + '-' + str(num)[:-2] - wav_title = title + '-' + role + '-' + cv_name + '-' + str(num) + title = add_zero_to_title(title[0], places) + roleName = row[selectedColumnList[2]] + linesNum = str(row[selectedColumnList[3]]) + if '.' in linesNum: + wav_title = title + '-' + roleName + '-' + cvName + '-' + linesNum[:-2] + elif linesNum == 'nan': + wav_title = title + '-' + roleName + '-' + cvName + '-' + '0' + else: + wav_title = title + '-' + roleName + '-' + cvName + '-' + linesNum wav_title_list.append(wav_title) - print(wav_title_list) + return wav_title_list -def creat_wav_file(wav_title_list): - # 设置音频参数 - sample_width = 3 # 24位深度为3个字节 - sample_rate = 44100 # 采样率为44100Hz - num_channels = 1 # 单声道 +def creat_wav_file(wav_title_list, sample_width, sample_rate, num_channels, save_path): + """ + :param wav_title_list: 波形文件名列表 + :param sample_width: 位深度，24位深度为3个字节 + :param sample_rate: 采样率 + :param num_channels: 声道，1为单声道，2为立体声 + :return: + """ # 创建WAV文件 for file_name in wav_title_list: - with wave.open(r"G:\ready\绝天神印\{}.wav".format(file_name), 'w') as wav_file: + wav_name = save_path + '/' + file_name + '.wav' + with wave.open(wav_name, 'w') as wav_file: # 设置音频参数 wav_file.setsampwidth(sample_width) wav_file.setframerate(sample_rate) @@ -59,11 +77,16 @@ def creat_wav_file(wav_title_list): wav_file.writeframes(sample) print('已成功创建{}文件'.format(file_name)) -if __name__ == '__main__': - file_path = r"C:\Users\Administrator\Downloads\aadd.xlsx" - column_list = ['章节名称', '角色名', '台词量'] - cv_name = '请叫我张钧' - wav_title_list = get_data(file_path, column_list, cv_name) - print(wav_title_list) - creat_wav_file(wav_title_list) +if __name__ == '__main__': + file_path = r"C:\Users\Cori\Downloads\逍遥小医仙剧组 (1).xlsx" + # column_list = ['章节名称', 'CV名称', '角色名', '台词量'] + column_list = ['章节名', 'CV', '角色名', '台词数'] + cv_name = '请叫我张钧' + sheetName = '角色音进度表201-487' + sample_width = 3 # 24位深度为3个字节 + sample_rate = 44100 # 采样率为44100Hz + num_channels = 1 # 单声道 + wav_title_list = get_wav_title(file_path, column_list, sheetName, cv_name, 4) + print(wav_title_list, sample_width, sample_rate, num_channels) + # creat_wav_file(wav_title_list)