0

xI have two fields I'm comparing. Both of them are nominals, with only ones and zeros. Men and women, and default and no default (plus lots more data in a similar format). If you put it into a scatter plot you only get four dots because of course all those scenarios happen, but the question is how many times does it happen in each case. It would be amazing if I could see it by flipping it on its side and see the frequency of how many times each dot hits one of those four points.

for example:

x = [1,0,0,0,...1,1,0,1]
y = [0,1,1,0,...1,0,1,0]

I have the code:

def scatterPlot3dFields():
    fig = plt.figure()
    ax = fig.add_subplot(111, projection='3d')
    ax.scatter(x, y, ???)
    plt.show()

But I have no idea what to put in the z axis to make this happen. Any help would be amazing.

user3023715
  • 1,539
  • 2
  • 11
  • 12

1 Answers1

0

It sounds like you're looking for np.histogram2d, which is a great method for determining the frequency (in the sense of number of occurences) for 2-dimensional data. You could try something like:

import numpy as np

# data of 1s and 0s
n_points = 100
x = np.random.randint(2, size=n_points) # ([1, 0, 1, ... , 0, 0, 1])
y = np.random.randint(2, size=n_points) # ([0, 1, 1, ... , 1, 0, 1])

H, xedges, yedges  = np.histogram2d(x, y, bins=2)
# H = frequency from the four points (0, 0), (0,1), (1,0), and (1,1)
# H = ([21, 26], [32, 21]]) for example (must sum to n_points (100))

It also sounds like you want to visualize the frequency. To do so, you can use a 3D bar graph (bar3d). Check out this matplotlib example code.

lanery
  • 5,222
  • 3
  • 29
  • 43