first commint
This commit is contained in:
suqin
256
read_excel.py
Normal file
256
read_excel.py
Normal file
@@ -0,0 +1,256 @@
|
||||
import os
|
||||
import pandas as pd
|
||||
from openpyxl import load_workbook
|
||||
|
||||
|
||||
def _merge_read_config(config):
|
||||
defaults = {
|
||||
"columns_to_preview": 3,
|
||||
"green_colors": {"FF92D050", "FF00B050"},
|
||||
"blue_colors": {"FF00B0F0", "FF0070C0"},
|
||||
"orange_colors": {"FFC55B11", "FFC55C10", "FFB45F06"},
|
||||
"pink_colors": {"FFEF949F", "FFFF99FF"},
|
||||
"log_file": "w.log",
|
||||
}
|
||||
if not config:
|
||||
return defaults
|
||||
for key, value in config.items():
|
||||
if value is not None:
|
||||
defaults[key] = value
|
||||
return defaults
|
||||
|
||||
|
||||
def _load_map_dataframe(file_path, sheet_name, start_i):
|
||||
if not os.path.exists(file_path):
|
||||
print(f"错误：文件未找到，请检查路径。Path: {file_path}")
|
||||
return None
|
||||
|
||||
try:
|
||||
df = pd.read_excel(file_path, sheet_name=sheet_name, engine="openpyxl")
|
||||
except Exception as e:
|
||||
print(f"读取 XLSX 文件时发生错误: {e}")
|
||||
if "No module named 'openpyxl'" in str(e):
|
||||
print("\n请确保您已经安装了 openpyxl 库：pip install openpyxl")
|
||||
return None
|
||||
|
||||
if len(df.columns) < 2:
|
||||
print("错误：表格至少需要两列 (地图编号与 Y 坐标)。")
|
||||
return None
|
||||
|
||||
map_col = df.columns[0]
|
||||
y_col = df.columns[1]
|
||||
grid_columns = df.columns[1:27]
|
||||
|
||||
i = start_i
|
||||
for row in df.itertuples():
|
||||
if pd.isna(row[1]):
|
||||
df.at[row.Index, map_col] = i
|
||||
else:
|
||||
i = row[1]
|
||||
|
||||
return df, map_col, y_col, grid_columns
|
||||
|
||||
|
||||
def _write_color_output(filename, rows, columns, success_msg, empty_msg):
|
||||
if os.path.exists(filename):
|
||||
os.remove(filename)
|
||||
if rows:
|
||||
pd.DataFrame(rows, columns=columns).to_excel(filename, index=False)
|
||||
print(success_msg)
|
||||
else:
|
||||
print(empty_msg)
|
||||
|
||||
|
||||
def find_cells_with_currency_xlsx(file_path, sheet_name="Sheet1", start_i=90091, search_term="币", config=None):
|
||||
"""
|
||||
读取 XLSX 文件，查找包含指定词语（默认为“币”）的单元格，
|
||||
并返回这些单元格的 (地图号, X, Y[, 方向])，结果同时保存为 found_xxx.xlsx。
|
||||
"""
|
||||
print(f"--- 正在处理文件: {os.path.basename(file_path)} ---")
|
||||
|
||||
df_info = _load_map_dataframe(file_path, sheet_name, start_i)
|
||||
if df_info is None:
|
||||
return []
|
||||
|
||||
cfg = _merge_read_config(config)
|
||||
|
||||
df, map_col, y_col, grid_columns = df_info
|
||||
results = []
|
||||
|
||||
# 2. 遍历 B~AA 每一列，在地图区域里搜索 币 / p / v
|
||||
for column in grid_columns:
|
||||
column_series = df[column].astype(str).fillna("")
|
||||
|
||||
if search_term in ["p", "v"]:
|
||||
# p/v 支持后缀方向
|
||||
suffixes = ["W", "S", "N", "E", "w", "s", "n", "e"]
|
||||
search_terms = [search_term] + [search_term + s for s in suffixes]
|
||||
matching_rows = column_series.apply(
|
||||
lambda x: any(term in x for term in search_terms)
|
||||
)
|
||||
else:
|
||||
search_terms = [search_term]
|
||||
matching_rows = column_series.str.contains(search_term, regex=False, na=False)
|
||||
|
||||
matching_indices = df[matching_rows].index.tolist()
|
||||
|
||||
if not matching_indices:
|
||||
continue
|
||||
|
||||
for idx in matching_indices:
|
||||
# 地图号和坐标：直接从 A、B 列取值
|
||||
map_no = df.at[idx, map_col] # A 列 = 地图编号
|
||||
y_no = df.at[idx, y_col] # B 列 = Y 坐标
|
||||
x_no = column # 列名本身 = X 坐标（-12 ~ 12）
|
||||
|
||||
# 提取方向（只对 p / v 有意义）
|
||||
direction = ""
|
||||
if search_term in ["p", "v"]:
|
||||
cell_text = str(df.loc[idx, column])
|
||||
direction = cell_text[len(search_term):] if len(cell_text) > len(search_term) else ""
|
||||
|
||||
if search_term in ["p", "v"]:
|
||||
results.append((map_no, x_no, y_no, direction))
|
||||
else:
|
||||
results.append((map_no, x_no, y_no))
|
||||
|
||||
# 4. 保存币 / p / v 结果到 Excel
|
||||
output_file = f"found_{search_term}.xlsx"
|
||||
if search_term in ["p", "v"]:
|
||||
results_df = pd.DataFrame(
|
||||
results,
|
||||
columns=["地图号 (Map No)", "X坐标 (X No)", "Y坐标 (Y No)", "方向 (Direction)"]
|
||||
)
|
||||
else:
|
||||
results_df = pd.DataFrame(
|
||||
results,
|
||||
columns=["地图号 (Map No)", "X坐标 (X No)", "Y坐标 (Y No)"]
|
||||
)
|
||||
|
||||
results_df.to_excel(output_file, index=False)
|
||||
print(f"\n🔖 已将结果保存到文件: {output_file}")
|
||||
|
||||
return results
|
||||
|
||||
|
||||
def scan_color_blocks(file_path, sheet_name="Sheet1", start_i=90091, config=None):
|
||||
"""扫描地图区域内的颜色格子，并把结果写入对应的 Excel 文件。"""
|
||||
|
||||
print("\n开始扫描颜色图案格子…")
|
||||
|
||||
df_info = _load_map_dataframe(file_path, sheet_name, start_i)
|
||||
if df_info is None:
|
||||
return {}
|
||||
|
||||
cfg = _merge_read_config(config)
|
||||
|
||||
df, map_col, y_col, grid_columns = df_info
|
||||
|
||||
try:
|
||||
wb = load_workbook(file_path, data_only=True)
|
||||
except Exception as e:
|
||||
print(f"读取 XLSX 文件时发生错误: {e}")
|
||||
return {}
|
||||
|
||||
try:
|
||||
ws = wb[sheet_name]
|
||||
except KeyError:
|
||||
active_sheet = wb.active
|
||||
print(f"未找到工作表 {sheet_name}，使用活动表 {active_sheet.title}。")
|
||||
ws = active_sheet
|
||||
|
||||
block_results = []
|
||||
water_results = []
|
||||
wall_results = []
|
||||
jump_results = []
|
||||
|
||||
excel_row_offset = 2 # DataFrame 行索引 0 对应 Excel 的第 2 行
|
||||
|
||||
import logging
|
||||
|
||||
logging.basicConfig(filename=cfg["log_file"], level=logging.INFO)
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
for idx in df.index:
|
||||
map_no = df.at[idx, map_col]
|
||||
y_val = df.at[idx, y_col]
|
||||
|
||||
try:
|
||||
y_int = int(y_val)
|
||||
except (TypeError, ValueError):
|
||||
continue
|
||||
|
||||
if y_int < -12 or y_int > 12:
|
||||
continue
|
||||
|
||||
excel_row = idx + excel_row_offset
|
||||
|
||||
for col in grid_columns[1:]: # 从 C 列开始（跳过 Y 轴列）
|
||||
col_idx = df.columns.get_loc(col) + 1 # openpyxl 的列号从 1 开始
|
||||
cell = ws.cell(row=excel_row, column=col_idx)
|
||||
fill = cell.fill
|
||||
|
||||
if fill is None or fill.patternType in (None, "none"):
|
||||
continue
|
||||
|
||||
fg = fill.fgColor
|
||||
rgb = getattr(fg, "rgb", None)
|
||||
index = getattr(fg, "index", None)
|
||||
color_code = rgb or index
|
||||
|
||||
if color_code is None:
|
||||
continue
|
||||
|
||||
c = str(color_code).upper()
|
||||
if c.startswith("FF"):
|
||||
log.info("Cell at Row %s, Column %s has color code: %s", excel_row, col_idx, c)
|
||||
|
||||
if c in cfg["green_colors"]:
|
||||
block_results.append((map_no, col, y_val))
|
||||
if c in cfg["blue_colors"]:
|
||||
water_results.append((map_no, col, y_val))
|
||||
if c in cfg["orange_colors"]:
|
||||
wall_results.append((map_no, col, y_val))
|
||||
if c in cfg["pink_colors"]:
|
||||
jump_results.append((map_no, col, y_val))
|
||||
|
||||
columns = ["地图号 (Map No)", "X坐标 (X No)", "Y坐标 (Y No)"]
|
||||
|
||||
_write_color_output(
|
||||
"found_方块.xlsx",
|
||||
block_results,
|
||||
columns,
|
||||
"🔖 已将所有绿色图案格子的坐标保存到文件: found_方块.xlsx",
|
||||
"未发现任何绿色图案格子。",
|
||||
)
|
||||
|
||||
_write_color_output(
|
||||
"found_water.xlsx",
|
||||
water_results,
|
||||
columns,
|
||||
"🔖 已将蓝色 (water) 格子保存到文件: found_water.xlsx",
|
||||
"未发现蓝色 (water) 格子。",
|
||||
)
|
||||
|
||||
_write_color_output(
|
||||
"found_wall.xlsx",
|
||||
wall_results,
|
||||
columns,
|
||||
"🔖 已将橙色 (wall) 格子保存到文件: found_wall.xlsx",
|
||||
"未发现橙色 (wall) 格子。",
|
||||
)
|
||||
|
||||
_write_color_output(
|
||||
"found_jump.xlsx",
|
||||
jump_results,
|
||||
columns,
|
||||
"🔖 已将粉色 (jump) 格子保存到文件: found_jump.xlsx",
|
||||
"未发现粉色 (jump) 格子。",
|
||||
)
|
||||
|
||||
return {
|
||||
"block": block_results,
|
||||
"water": water_results,
|
||||
"wall": wall_results,
|
||||
"jump": jump_results,
|
||||
}
|
||||
Reference in New Issue
Block a user