Skip to main content

The ForceAtlas2 algorithm for Python (and NetworkX)

Project description

This is a port of Gephi’s ForceAtlas2 layout algorithm to Python 2 and Python 3 (with a wrapper for NetworkX).

Installation

The only strict dependency is Numpy. Cython is highly recommended, as it will speed it up by a factor of 10-100, depending on the graph. Both can be installed with:

pip install numpy pip install cython

To use the NetworkX wrapper function, you obviously need NetworkX.

Then to install this package, it is the typical:

python setup.py install

Example usage

Two core functions are defined:

  • forceatlas2.forceatlas2 takes an adjacency matrix and optionally initial positions and other optional arguments. It uses this to calculate a layout.

  • forceatlas2.forceatlas2_networkx_layout takes a NetworkX graph and returns a NetworkX-compatible layout.

To use within NetworkX:

import matplotlib.pyplot as plt import networkx import random import forceatlas2 G = networkx.karate_club_graph() pos = { i : (random.random(), random.random()) for i in G.nodes()} # Optionally specify positions as a dictionary l = forceatlas2.forceatlas2_networkx_layout(G, pos, niter=1000) # Optionally specify iteration count networkx.draw_networkx(G, l) plt.show()

Note that my coding style is non-Pythonic, as I don’t use function docstrings. You will have to take a look at the function definition in forceatlas2.py to see a description of all of the potential arguments.

Project details


Release history Release notifications | RSS feed

This version

1.0

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

ForceAtlas2-1.0.tar.gz (6.8 kB view details)

Uploaded Source

File details

Details for the file ForceAtlas2-1.0.tar.gz.

File metadata

  • Download URL: ForceAtlas2-1.0.tar.gz
  • Upload date:
  • Size: 6.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No

File hashes

Hashes for ForceAtlas2-1.0.tar.gz
Algorithm Hash digest
SHA256 fdc3d3ea7f85b6535598c30ced3b5ccc7b1ac2e49dc486f4a19c6b9fa998a81d
MD5 2155b891eadc6322528e1f6313bd95c9
BLAKE2b-256 17401966ff5370616809a400085c2e782c0b5f93ed838e437eb2371bc3033414

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page