62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
|
"""Aggiunge ai CSV dell'ultimo ciclo una colonna di potenziale normalizzato.
|
||
|
|
|
||
|
|
Chiede all'utente il potenziale di riferimento (in V) e aggiunge, per ogni
|
||
|
|
file *_ultimo_ciclo.csv nella cartella, una terza colonna pari a
|
||
|
|
Potential/V - riferimento. I file vengono sovrascritti.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import csv
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
INPUT_SUFFIX = "_ultimo_ciclo"
|
||
|
|
|
||
|
|
|
||
|
|
def ask_reference_potential() -> float:
|
||
|
|
while True:
|
||
|
|
raw = input("Potenziale di riferimento (V): ").strip()
|
||
|
|
try:
|
||
|
|
return float(raw)
|
||
|
|
except ValueError:
|
||
|
|
print("Valore non valido, inserisci un numero (es. -0.152).")
|
||
|
|
|
||
|
|
|
||
|
|
def normalize_file(path: Path, reference: float) -> None:
|
||
|
|
with path.open("r", encoding="utf-8", newline="") as f:
|
||
|
|
reader = csv.reader(f, delimiter=";")
|
||
|
|
header = next(reader)
|
||
|
|
rows = [(row[0], row[1]) for row in reader]
|
||
|
|
|
||
|
|
header = [*header, "Potential normalized/V"]
|
||
|
|
new_rows = []
|
||
|
|
for potential_str, current_str in rows:
|
||
|
|
potential = float(potential_str.replace(",", "."))
|
||
|
|
normalized = potential - reference
|
||
|
|
new_rows.append((potential_str, current_str, f"{normalized:.3f}".replace(".", ",")))
|
||
|
|
|
||
|
|
with path.open("w", encoding="utf-8", newline="") as f:
|
||
|
|
writer = csv.writer(f, delimiter=";")
|
||
|
|
writer.writerow(header)
|
||
|
|
writer.writerows(new_rows)
|
||
|
|
|
||
|
|
|
||
|
|
def main(argv: list[str]) -> None:
|
||
|
|
folder = Path(argv[1]) if len(argv) > 1 else Path(__file__).parent
|
||
|
|
csv_files = sorted(
|
||
|
|
p for p in folder.glob("*.csv") if p.stem.endswith(INPUT_SUFFIX)
|
||
|
|
)
|
||
|
|
|
||
|
|
if not csv_files:
|
||
|
|
print(f"Nessun file *{INPUT_SUFFIX}.csv trovato in {folder}")
|
||
|
|
return
|
||
|
|
|
||
|
|
reference = ask_reference_potential()
|
||
|
|
|
||
|
|
for csv_file in csv_files:
|
||
|
|
normalize_file(csv_file, reference)
|
||
|
|
print(f"OK {csv_file.name} (potenziale normalizzato aggiunto)")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main(sys.argv)
|