46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
|
import pandas as pd
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import random
|
||
|
|
|
||
|
|
class Automat:
|
||
|
|
def __init__(self, df: pd.DataFrame, end_states: list):
|
||
|
|
self._df = df
|
||
|
|
self._end_states = end_states
|
||
|
|
|
||
|
|
def _consume_character(self, state: int, char: str):
|
||
|
|
filter = (self._df[0] == state) & (self._df[2] == char)
|
||
|
|
matching_rows = self._df[filter]
|
||
|
|
if matching_rows.empty:
|
||
|
|
return None
|
||
|
|
end_states = matching_rows.iloc[:,1].tolist()
|
||
|
|
return random.choice(end_states)
|
||
|
|
|
||
|
|
def consume_word(self, word: str):
|
||
|
|
prev_state = 0
|
||
|
|
state = 0
|
||
|
|
for char in word:
|
||
|
|
prev_state = state
|
||
|
|
state = self._consume_character(state, char)
|
||
|
|
if state is None:
|
||
|
|
return prev_state in self._end_states
|
||
|
|
return state in self._end_states
|
||
|
|
|
||
|
|
def create_automat_from_file(file_path: str):
|
||
|
|
automat_df = pd.read_csv(file_path, sep=' ', header=None)
|
||
|
|
end_states_condition = (automat_df[1].isnull()) & (automat_df[2].isnull())
|
||
|
|
end_states_df = automat_df[end_states_condition]
|
||
|
|
end_states = end_states_df.iloc[:,0].tolist()
|
||
|
|
|
||
|
|
automat_df = automat_df[~end_states_condition]
|
||
|
|
automat_df[2] = automat_df[2].astype(str)
|
||
|
|
|
||
|
|
return Automat(automat_df, end_states)
|
||
|
|
|
||
|
|
automat_path = os.path.join(os.path.dirname(__file__), 'test.arg')
|
||
|
|
automat = create_automat_from_file(automat_path)
|
||
|
|
|
||
|
|
txt = input('Write valid input: ')
|
||
|
|
while txt != 'exit':
|
||
|
|
print('YES' if automat.consume_word(txt) else 'NO')
|
||
|
|
txt = input('Write valid input: ')
|