ciphers/polybius.py · python/Python

Fetch the repository succeeded.

richgiteeai

Watch

Unwatch Watching Releases Only Ignoring

1 Star 0 Fork 324

python/Python

forked from 编程语言算法集/Python

Code Issues 0 Pull Requests 0 Wiki Insights Pipelines

Service

Tencent Cloud Serverless

Create your Gitee Account

Explore and code with more than 14 million developers,Free private repositories !:)

Already have an account? Sign in

文件

master

Branches (54)

Manage

master

mypy-reality-check

Python-3.10-beta-1

bubble_sort

Fix-validate_initial_digits-of-credit_card_validator.py

long-running-tests

fix-project-euler-problem-15

mypy-project-euler-092-sol-1

Add-tests-to-morse_code.py

from-__future__-import-annotations

flake8-bugbear

is_ip_v4_address_valid.py

cclauss-patch-4

cclauss-patch-3

mypy_--install-types_--non-interactive

Write-for-current-Python

harris_corner.py

dhruvmanila-patch-1

refactor/action

fix/action

Clone or Download

Clone/Download

HTTPS SSH SVN SVN+SSH Download ZIP

Prompt

To download the code, please copy the following command and execute it in the terminal

To ensure that your submitted code identity is correctly recognized by Gitee, please execute the following command.

git config --global user.name userName 
git config --global user.email userEmail

When using the SSH protocol for the first time to clone or push code, follow the prompts below to complete the SSH configuration.

1 Generate RSA keys.

2 Obtain the content of the RSA public key and configure it in SSH Public Keys

To use SVN on Gitee, please visit the usage guide

When using the HTTPS protocol, the command line will prompt for account and password verification as follows. For security reasons, Gitee recommends configure and use personal access tokens instead of login passwords for cloning, pushing, and other operations.

Username for 'https://gitee.com': userName

Password for 'https://userName@gitee.com': # Private Token

Branches 54

Tags 0

Python

ciphers

polybius.py

polybius.py 3.07 KB

#!/usr/bin/env python3

"""
A Polybius Square is a table that allows someone to translate letters into numbers.

https://www.braingle.com/brainteasers/codes/polybius.php
"""

import numpy as np

class PolybiusCipher:
 def __init__(self) -> None:
 SQUARE = [
 ["a", "b", "c", "d", "e"],
 ["f", "g", "h", "i", "k"],
 ["l", "m", "n", "o", "p"],
 ["q", "r", "s", "t", "u"],
 ["v", "w", "x", "y", "z"],
 ]
 self.SQUARE = np.array(SQUARE)

def letter_to_numbers(self, letter: str) -> np.ndarray:
 """
 Return the pair of numbers that represents the given letter in the
 polybius square
 >>> np.array_equal(PolybiusCipher().letter_to_numbers('a'), [1,1])
 True

>>> np.array_equal(PolybiusCipher().letter_to_numbers('u'), [4,5])
 True
 """
 index1, index2 = np.where(self.SQUARE == letter)
 indexes = np.concatenate([index1 + 1, index2 + 1])
 return indexes

def numbers_to_letter(self, index1: int, index2: int) -> str:
 """
 Return the letter corresponding to the position [index1, index2] in
 the polybius square

>>> PolybiusCipher().numbers_to_letter(4, 5) == "u"
 True

>>> PolybiusCipher().numbers_to_letter(1, 1) == "a"
 True
 """
 return self.SQUARE[index1 - 1, index2 - 1]

def encode(self, message: str) -> str:
 """
 Return the encoded version of message according to the polybius cipher

>>> PolybiusCipher().encode("test message") == "44154344 32154343112215"
 True

>>> PolybiusCipher().encode("Test Message") == "44154344 32154343112215"
 True
 """
 message = message.lower()
 message = message.replace("j", "i")

encoded_message = ""
 for letter_index in range(len(message)):
 if message[letter_index] != " ":
 numbers = self.letter_to_numbers(message[letter_index])
 encoded_message = encoded_message + str(numbers[0]) + str(numbers[1])
 elif message[letter_index] == " ":
 encoded_message = encoded_message + " "

return encoded_message

def decode(self, message: str) -> str:
 """
 Return the decoded version of message according to the polybius cipher

>>> PolybiusCipher().decode("44154344 32154343112215") == "test message"
 True

>>> PolybiusCipher().decode("4415434432154343112215") == "testmessage"
 True
 """
 message = message.replace(" ", " ")
 decoded_message = ""
 for numbers_index in range(int(len(message) / 2)):
 if message[numbers_index * 2] != " ":
 index1 = message[numbers_index * 2]
 index2 = message[numbers_index * 2 + 1]

letter = self.numbers_to_letter(int(index1), int(index2))
 decoded_message = decoded_message + letter
 elif message[numbers_index * 2] == " ":
 decoded_message = decoded_message + " "

return decoded_message

Copy Edit Raw Blame History

Erwin Junge authored 2021年10月26日 18:35 +08:00 . [mypy] annotate ciphers (#5569)

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95

#!/usr/bin/env python3

"""
A Polybius Square is a table that allows someone to translate letters into numbers.

https://www.braingle.com/brainteasers/codes/polybius.php
"""

import numpy as np


class PolybiusCipher:
 def __init__(self) -> None:
 SQUARE = [
 ["a", "b", "c", "d", "e"],
 ["f", "g", "h", "i", "k"],
 ["l", "m", "n", "o", "p"],
 ["q", "r", "s", "t", "u"],
 ["v", "w", "x", "y", "z"],
 ]
 self.SQUARE = np.array(SQUARE)

 def letter_to_numbers(self, letter: str) -> np.ndarray:
 """
 Return the pair of numbers that represents the given letter in the
 polybius square
 >>> np.array_equal(PolybiusCipher().letter_to_numbers('a'), [1,1])
 True

 >>> np.array_equal(PolybiusCipher().letter_to_numbers('u'), [4,5])
 True
 """
 index1, index2 = np.where(self.SQUARE == letter)
 indexes = np.concatenate([index1 + 1, index2 + 1])
 return indexes

 def numbers_to_letter(self, index1: int, index2: int) -> str:
 """
 Return the letter corresponding to the position [index1, index2] in
 the polybius square

 >>> PolybiusCipher().numbers_to_letter(4, 5) == "u"
 True

 >>> PolybiusCipher().numbers_to_letter(1, 1) == "a"
 True
 """
 return self.SQUARE[index1 - 1, index2 - 1]

 def encode(self, message: str) -> str:
 """
 Return the encoded version of message according to the polybius cipher

 >>> PolybiusCipher().encode("test message") == "44154344 32154343112215"
 True

 >>> PolybiusCipher().encode("Test Message") == "44154344 32154343112215"
 True
 """
 message = message.lower()
 message = message.replace("j", "i")

 encoded_message = ""
 for letter_index in range(len(message)):
 if message[letter_index] != " ":
 numbers = self.letter_to_numbers(message[letter_index])
 encoded_message = encoded_message + str(numbers[0]) + str(numbers[1])
 elif message[letter_index] == " ":
 encoded_message = encoded_message + " "

 return encoded_message

 def decode(self, message: str) -> str:
 """
 Return the decoded version of message according to the polybius cipher

 >>> PolybiusCipher().decode("44154344 32154343112215") == "test message"
 True

 >>> PolybiusCipher().decode("4415434432154343112215") == "testmessage"
 True
 """
 message = message.replace(" ", " ")
 decoded_message = ""
 for numbers_index in range(int(len(message) / 2)):
 if message[numbers_index * 2] != " ":
 index1 = message[numbers_index * 2]
 index2 = message[numbers_index * 2 + 1]

 letter = self.numbers_to_letter(int(index1), int(index2))
 decoded_message = decoded_message + letter
 elif message[numbers_index * 2] == " ":
 decoded_message = decoded_message + " "

 return decoded_message