FilesExpand file tree

 main

/

test_bytes_numPy.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

21 lines (12 loc) · 594 Bytes

 main

/

test_bytes_numPy.py

Copy path

Top

File metadata and controls

• Code
• Blame

21 lines (12 loc) · 594 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

import numpy as np

payload = b'ABCAA'

if len(payload) == 0:

exit

counts = np.bincount(np.frombuffer(payload, dtype=np.uint8), minlength=256)

probabilities = counts / len(payload)

probabilities = probabilities[probabilities > 0]

log2array = np.log2(probabilities)

log2probsArray = probabilities * np.log2(probabilities)

entropy_val = -np.sum(probabilities * np.log2(probabilities))

square_entropy = np.nansum((1 - np.bincount(np.frombuffer(payload, dtype=np.uint8)) / len(payload)) * np.sqrt(np.bincount(np.frombuffer(payload, dtype=np.uint8)) / len(payload)))

print(square_entropy)